from django.conf import settings

from django.utils.datastructures import SortedDict

import uuid

DATE_FORMATS = ["%d/%m/%Y", "%Y-%m-%d"]

class SegmentSerializer(object):

    """

    Serialize a set of annotations to a cinelab compatible array

    """

    def __init__(self, content, segments, viewable_contents=[], default_color=2194379):

        self.content = content

        self.segments = segments

        self.viewable_contents = viewable_contents

        self.default_color = default_color

        self.views = None

        self.annotation_types = None

        self.medias = None

        self.annotations = None

    def __parse_views(self):

        view = {

                "id": "0",

                "contents": [

                    self.content.iri_id

                ],

                "annotation_types": [ ],

        }

        self.annotation_types = []

        annotation_types = [seg.cutting_id for seg in self.segments]

        for a in annotation_types:

            view['annotation_types'].append(a)

            self.annotation_types.append({

                    "dc:contributor": "undefined",

                    "dc:creator": "undefined",

                    "dc:title": "cutting %s" % a,

                    "id": "%s" % a,

                    "dc:created": "",

                    "dc:description": "",

                    "dc:modified": ""

                })

        self.views = [view]

    def __parse_content(self):

        href = ""

        meta_item_value = ""

        if self.content.iri_id not in self.viewable_contents:

            href = settings.FORBIDDEN_STREAM_URL

        elif self.content.videopath:

            href = self.content.videopath.rstrip('/') + "/" + self.content.src

            meta_item_value = self.content.videopath.rstrip('/') + "/"

        else:

            href = self.content.src

        media = {

             "http://advene.liris.cnrs.fr/ns/frame_of_reference/ms" : "o=0",

             "id" : self.content.iri_id,

             "href" : href,

             "unit" : "ms",

             "origin" : "0",

             "meta": {

                 "dc:creator" : "",

                 "dc:created" : self.content.creation_date.isoformat(),

                 "dc:contributor" : "",

                 "dc:modified" : self.content.update_date.isoformat(),

                 "dc:creator.contents" : "",

                 "dc:title" : self.content.title,

                 "dc:description" : self.content.description,

                 "dc:duration" : self.content.get_duration(),

                 "item": {

                     "name" : "streamer",

                     "value": meta_item_value,

                 },

             }

        }

        self.medias = [media]

    def __parse_segments(self):

        self.annotations = []

        for seg in self.segments:

            segment = {

                'begin': seg.start_ts,

                'end': seg.start_ts + seg.duration,

                'tags': seg.tags,

                'id': "s_%s" % unicode(uuid.uuid1()),

                'color': "%s" % self.default_color,

                'media': self.content.iri_id,

                'content': {

                    'mimetype': 'application/x-ldt-structured',

                    'description': seg.abstract,

                    'img': {

                        'src': ''

                    },

                    'title': seg.title,

                    'color': '2194379',

                    'polemics': [ ],

                    'audio': {

                        'mimetype': 'audio/mp3',

                        'src': '',

                        'href': 'null'

                    }

                },

                'meta': {

                         "dc:creator": seg.author,

                         "dc:contributor": seg.author,

                         "dc:created": seg.date,

                         "dc:modified": seg.date,

                         "id-ref": seg.iri_id,

                         "project": seg.project_id

                         }

            }

            self.annotations.append(segment)

    def serialize_to_cinelab(self):

        self.__parse_content()

        self.__parse_segments()

        self.__parse_views()

        res = {}

        res['views'] = self.views 

        res['tags'] = None

        res['lists'] = None

        res['medias'] = self.medias

        res['annotations'] = self.annotations

        res['annotation-types'] = self.annotation_types

        return res