From: Guillaume Pellerin <yomguy@parisson.com>
Date: Mon, 14 Oct 2013 23:55:34 +0000 (+0200)
Subject: add item import command
X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=e12b2340411e9428d009b02c6931f7114910d541;p=telemeta.git

add item import command
---

diff --git a/telemeta/management/commands/__init__.py b/telemeta/management/commands/__init__.py
index e69de29b..40a96afc 100644
--- a/telemeta/management/commands/__init__.py
+++ b/telemeta/management/commands/__init__.py
@@ -0,0 +1 @@
+# -*- coding: utf-8 -*-
diff --git a/telemeta/management/commands/telemeta-import-items.py b/telemeta/management/commands/telemeta-import-items.py
new file mode 100644
index 00000000..9ab3c4b5
--- /dev/null
+++ b/telemeta/management/commands/telemeta-import-items.py
@@ -0,0 +1,41 @@
+from optparse import make_option
+from django.conf import settings
+from django.core.management.base import BaseCommand, CommandError
+from django.core.files.base import ContentFile
+from telemeta.models import *
+from telemeta.util.unaccent import unaccent
+import os
+
+
+class Command(BaseCommand):
+    help = "import media files from a directory to a collection"
+    args = "collection_code media_dir"
+
+    def handle(self, *args, **options):
+        collection_code = args[-2]
+        media_dir = args[-1]
+
+        collections = MediaCollection.objects.filter(code=collection_code)
+        if not collections:
+            collection = MediaCollection(code=collection_code, title=collection_code)
+            collection.public_access = 'full'
+            collection.save()
+            print 'collection created: ' + collection_code
+        else:
+            collection = collections[0]
+            print 'using collection: ' + collection.code
+
+        for root, dirs, files in os.walk(media_dir):
+            for filename in files:
+                path = root + os.sep + filename
+                items = MediaItem.objects.filter(code=filename)
+                if not items:
+                    item = MediaItem(collection=collection, code=filename)
+                    f = open(path, 'r')
+                    file_content = ContentFile(f.read())
+                    item.file.save(filename, file_content)
+                    item.public_access = 'full'
+                    item.save()
+                    print 'item created: ' + item.code
+                else:
+                    print 'item already exists: ' + item.code