'''

Imports amateur.iri.centrepompidou.fr/nouveaumonde files to ldt.iri

'''

import csv

import sys

import requests

from django.core.management.base import BaseCommand

from django.core import management

from lxml import etree

from django.db import connections

from ldt.ldt_utils import models

def create_source(iri_id):

    '''

    create .iri source

    '''

    return "https://media.iri.centrepompidou.fr/video/enmi/" + iri_id + ".mp4"

def create_iri_url(iri_url):

    '''

    create iri url

    '''

    return "http://amateur.iri.centrepompidou.fr" + iri_url

class Command(BaseCommand):

    '''

    command

    '''

    def add_arguments(self, parser):

        '''

        add arguments

        '''

        parser.add_argument('-i',

                            '--pathin',

                            dest='pathin',

                            default=None

                           )

        parser.add_argument('-o',

                            '--pathout',

                            dest='pathout',

                            default=None

                           )

    def get_duration(self, elem):

        '''

        get duration

        '''

        for element in self.mylist:

            element[0] = element[0][:len(element[0])-1]

            if elem == element[0]:

                element[1] = element[1][1:]

                root = etree.XML(element[1].encode('utf-8'), self.parser)

                duration = root.xpath('format')[0].get('duration')

                duration = duration[:7]

                return duration

    def change_annotations(self, iriin, ldtout):

        '''

        change annotations

        '''

        rootout = etree.XML(ldtout.encode('utf-8'), self.parser)

        rootin = etree.XML(iriin.encode('utf-8'), self.parser)

        ensembles = rootin.xpath('body/ensembles/ensemble')

        for ensemble in ensembles:

            idens = ensemble.get("id")

            decoups = ensemble.xpath('decoupage')

            rootout.xpath('annotations/content')[0].append(ensemble)

            for decoup in decoups:

                iddec = decoup.get("id")

                nouveaudecoup = '<decoupage idens=\"' + idens +'\" id=\"' + iddec +'\" tagsSelect=\"\"/>'

                nouveaudecxml = etree.fromstring(nouveaudecoup)

                rootout.xpath('displays/display/content')[0].append(nouveaudecxml)

        return etree.tostring(rootout)

    def handle(self, *args, **options):

        '''

        handle command

        '''

        pathin = options['pathin']

        pathout = options['pathout']

        if not pathin or not pathout:

            return "Error : specify path in and path out"

        try:

            csvfile = open(pathin, 'r')

            csvfile2 = open(pathout, 'wb')

        except IOError:

            self.stdout.write('file can\'t be opened')

            return

        cursor = connections['default2'].cursor()

        cursor.execute('SELECT ct.iri_id, ct.iriurl, ct.creation_date, ct.title, ct.description,\

                        pj.ldt_id, pj.ldt\

                        FROM ldt_content AS ct INNER JOIN ldt_ldtproject_contents\

                        AS ctpj ON ct.id = ctpj.content_id\

                        INNER JOIN ldt_ldtproject AS pj ON ctpj.ldtproject_id = pj.id;'

                      )

        amateurdata = cursor.fetchall()

        reload(sys)

        sys.setdefaultencoding('utf8')

        self.parser = etree.XMLParser(encoding='utf-8')

        self.myfile = csv.reader(csvfile)

        self.writefile = csv.writer(csvfile2)

        self.mylist = list(self.myfile)

        for mediaproj in amateurdata:

            iri_id = mediaproj[0]

            try:

                models.Project.objects.get(title='front project : %s' % iri_id)

                continue

            except models.Project.MultipleObjectsReturned:

                continue

            except models.Project.DoesNotExist:

                self.stdout.write('Media %s will be created'%iri_id)

            iriurl = mediaproj[1]

            #TODO set creationdate, title...

            # creationdate = mediaproj[2]

            # title = mediaproj[3]

            description = mediaproj[4]

            # ldt_id = mediaproj[5]

            # ldt = mediaproj[6]

            mysource = create_source(iri_id)

            myiriurl = create_iri_url(iriurl)

            myiri = requests.get(myiriurl)._content

            duration = self.get_duration(mysource)

            if requests.head(mysource).status_code == 200:

                management.call_command(

                    'createmediacontent',

                    source=mysource,

                    title=iri_id,

                    videopath='',

                    description=description,

                    duration=duration,

                    public=True,

                    creator='admin',

                    update=True

                    )

                myfrontproj = models.Project.objects.get(title='front project : %s' % iri_id)

                myfrontproj.ldt = self.change_annotations(myiri, myfrontproj.ldt)

                self.writefile.writerow([mysource,

                                         iri_id,

                                         models.Content.objects.get(title=iri_id).iri_id,

                                         myfrontproj.ldt_id

                                        ])

                self.stdout.write("Project changed")

                myfrontproj.save()

        csvfile.close()

        csvfile2.close()