add collection related media

diff --git a/telemeta/management/commands/telemeta-import-corpus-from-dir.py b/telemeta/management/commands/telemeta-import-corpus-from-dir.py
index c6322eb546b07d40e6650437fabd3537960ee54b..9625fa8f2d324f4393fd496550f415233782b819 100644 (file)
--- a/telemeta/management/commands/telemeta-import-corpus-from-dir.py
+++ b/telemeta/management/commands/telemeta-import-corpus-from-dir.py
@@ -114,7 +114,11 @@ class Command(BaseCommand):
                             metadata[data[0]] = data[1:]
                         i += 1
                     print metadata
-                    break
+
+                if os.path.isfile(path) and '.jpg' == os.path.splitext(filename)[1]:
+                    related_path = path.replace(self.media_root, '')
+                    related, c = MediaCollectionRelated.objects.get_or_create(collection=collection,
+                                    file=related_path)
 
             for root, dirs, files in os.walk(chapter_dir):
                 for media_file in files: