From: Guillaume Pellerin <yomguy@parisson.com>
Date: Tue, 31 Dec 2013 09:20:55 +0000 (+0100)
Subject: fix get collection and item
X-Git-Tag: 2.8.1-pro~343
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=9db73279481fb6fa0127cf36fbc7cfcb2431bea2;p=teleforma.git

fix get collection and item
---

diff --git a/teleforma/management/commands/teleforma-import-seminar-media.py b/teleforma/management/commands/teleforma-import-seminar-media.py
index 2eaa046c..9ffbf759 100644
--- a/teleforma/management/commands/teleforma-import-seminar-media.py
+++ b/teleforma/management/commands/teleforma-import-seminar-media.py
@@ -157,28 +157,13 @@ class Command(BaseCommand):
                     if not exist:
                         logger.logger.info(seminar.public_url())
                         logger.logger.info(path)
-                        collections = MediaCollection.objects.filter(code=collection_id)
-                        if not collections:
-                            collection = MediaCollection(code=collection_id,title=collection_id)
-                            collection.save()
-                        else:
-                            collection = collections[0]
-
-                        id = '_'.join([period.name, collection_id, ext, str(media_rank)])
-
-                        items = MediaItem.objects.filter(collection=collection, code=id)
-                        if not items:
-                            item = MediaItem(collection=collection, code=id)
-                            item.save()
-                        else:
-                            item = items[0]
-
+                        collections = MediaCollection.objects.get_or_create(code=collection_id)
+                        item_id = '_'.join([period.name, collection_id, ext, str(media_rank)])
+                        items = MediaItem.objects.get_or_create(collection=collection, code=item_id)
                         item.title = name
                         item.file = path
-
                         if os.path.getsize(root+os.sep+filename):
                             item.approx_duration = self.get_duration(root+os.sep+filename)
-
                         item.save()
 
                         files = os.listdir(root)