Merge branch 'dev'

Author: yomguy

teleforma/management/commands/teleforma-import-conferences-create.py

@@ -0,0 +1,173 @@
+import datetime
+from optparse import make_option
+from django.conf import settings
+from django.core.management.base import BaseCommand, CommandError
+from django.contrib.auth.models import User
+from django.template.defaultfilters import slugify
+from telemeta.models import *
+from telemeta.util.unaccent import unaccent
+from teleforma.models import *
+import logging
+import os
+#import timeside
+
+
+class Logger:
+    """A logging object"""
+
+    def __init__(self, file):
+        self.logger = logging.getLogger('myapp')
+        self.hdlr = logging.FileHandler(file)
+        self.formatter = logging.Formatter('%(asctime)s %(levelname)s %(message)s')
+        self.hdlr.setFormatter(self.formatter)
+        self.logger.addHandler(self.hdlr)
+        self.logger.setLevel(logging.INFO)
+
+
+class Command(BaseCommand):
+    help = "Import conferences from the MEDIA_ROOT directory, create conferences if don't exist"
+    admin_email = '[email protected]'
+    args = 'organization log_file'
+    spacer = '_-_'
+    original_format = ['webm', 'mp4']
+    transcoded_formats = ['mp4', 'ogg', 'mp3']
+    image_formats = ['png', 'jpg']
+
+    def cleanup(self):
+        medias = Media.objects.all()
+        for media in medias:
+            media.delete()
+        items = MediaItem.objects.all()
+        for item in items:
+            item.delete()
+
+    def get_duration(self, file):
+        #TODO: avoid timeside
+        #decoder = timeside.decoder.FileDecoder(file)
+        #value = str(period_nametime.timedelta(0,decoder.input_duration))
+        #t = value.split(':')
+        #t[2] = t[2].split('.')[0]
+        #return ':'.join(t)
+        return
+
+
+        #decoder.setup()
+        # time.sleep(0.5)
+    def handle(self, *args, **options):
+        organization_name = args[0]
+        department_name = args[1]
+        period_name = args[2]
+        log_file = args[3]
+        logger = Logger(log_file)
+
+        organization = Organization.objects.get(name=organization_name)
+        department = Department.objects.get(name=department_name,
+                                            organization=organization)
+        self.media_dir = settings.MEDIA_ROOT + organization.name + os.sep + department_name
+        file_list = []
+        all_conferences = Conference.objects.all()
+        i = 1
+
+#        self.cleanup()
+
+        for root, dirs, files in os.walk(self.media_dir):
+            for filename in files:
+                name = os.path.splitext(filename)[0]
+                ext = os.path.splitext(filename)[1][1:]
+
+                if ext and (ext in self.original_format or ext in self.transcoded_formats) and name[0] != '.':
+                    root_list = root.split(os.sep)
+                    public_id = root_list[-1]
+                    course = root_list[-2]
+                    course_code = course.split(self.spacer)[0]
+                    course_type_name = course.split(self.spacer)[1].lower()
+                    year = root_list[-3]
+                    department_name = root_list[-4]
+                    organization_name = root_list[-5]
+                    abs_path = root + os.sep + filename
+                    dir = os.sep.join(root_list[-5:])
+                    path = dir + os.sep + filename
+                    collection_id = '_'.join([department_name, course_code, course_type_name])
+                    
+                    print(public_id)
+                    courses = Course.objects.filter(code=course_code)
+                    course_type_obj = CourseType.objects.get(name=course_type_name)
+                    period_obj = Period.objects.get(name=period_name)
+                    mtime = os.path.getmtime(abs_path)
+                    conf_datetime = datetime.datetime.fromtimestamp(mtime)
+
+                    if department and courses:
+                        course_obj = courses[0]
+                        conferences = Conference.objects.filter(public_id=public_id)
+                        if conferences:
+                            conference = conferences[0]
+                        else:
+                            conference = Conference(public_id=public_id)
+                            conference.course = course_obj
+                            conference.course_type = course_type_obj
+                            conference.period = period_obj
+                            conference.date_begin = conf_datetime
+                            conference.save()
+
+                        department = Department.objects.get(name=department_name,
+                                                            organization=organization)
+                        exist = False
+                        medias = conference.media.all()
+                        for media in medias:
+                            if media.item.file == path:
+                                exist = True
+                                break
+
+                        streaming = False
+                        try:
+                            stations = conference.station.filter(started=True, public_id=public_id)
+                            if stations:
+                                streaming = True
+                        except:
+                            pass
+
+                        if not exist and not streaming:
+                            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([collection_id, public_id, ext, str(i)])
+
+                            items = MediaItem.objects.filter(collection=collection, code=id)
+                            if not items:
+                                item = MediaItem(collection=collection, code=id)
+                                item.save()
+                            else:
+                                item = items[0]
+
+                            item.title = name
+                            item.file = path
+                            #item.approx_duration = self.get_duration(root+os.sep+filename)
+                            item.save()
+
+                            files = os.listdir(root)
+                            for file in files:
+                                filename, extension = os.path.splitext(file)
+                                if extension[1:] in self.image_formats:
+                                    related = MediaItemRelated(item=item)
+                                    related.file = dir + os.sep + file
+                                    related.title = 'preview'
+                                    related.set_mime_type()
+                                    related.save()
+                                    break
+
+                            media = Media(conference=conference)
+                            media.item = item
+                            media.course = conference.course
+                            media.period = conference.period
+                            media.course_type = conference.course_type
+                            media.type = ext
+                            media.set_mime_type()
+                            media.save()
+                            conference.save()
+                            logger.logger.info(path)
+                            i += 1
+