X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/3d17368f18b2ca9696c5c2e4e431f7a230d1940b..469775ecf6cef4426cd6a89a84889a9c007b9d01:/catalogue/management/commands/importlessons.py

diff --git a/catalogue/management/commands/importlessons.py b/catalogue/management/commands/importlessons.py
index d179b16..d4733d3 100755
--- a/catalogue/management/commands/importlessons.py
+++ b/catalogue/management/commands/importlessons.py
@@ -33,6 +33,8 @@ class Command(BaseCommand):
     @staticmethod
     def all_attachments(path):
         files = {}
+        if not os.path.isdir(path):
+            return files
 
         def read_dir(path):
             for name in os.listdir(path):
@@ -94,6 +96,12 @@ class Command(BaseCommand):
             file_path = os.path.join(abs_dir, file_name)
             file_base, ext = os.path.splitext(file_path)
 
+            if os.path.isdir(file_path):
+                dir_imported, dir_skipped = self.import_from_dir(file_path)
+                files_imported += dir_imported
+                files_skipped += files_skipped
+                continue
+
             # Skip files that are not XML files
             if not ext == '.xml':
                 continue