from annotindexer import AnnotIndexer

from django.db import models

from django.utils.translation import ugettext_lazy as _

from tagging.models import Tag

from utils import generate_uuid

import ldt.indexation

import lucene

import lxml

import tagging.fields

#from django.core.management.validation import max_length

def Property(func):

    return property(**func()) 

class Annotation(models.Model):

    external_id = models.CharField(max_length=1024, null=False, unique=True, default=generate_uuid, verbose_name=_('annotation.external_id'))

    uri = models.CharField(max_length=1024, verbose_name=_('annotation.uri'))

    tags_field = tagging.fields.TagField(max_length=2048, null=True, blank=True, verbose_name=_('annotation.tags'))

    title = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('annotation.title'))

    description = models.TextField(null=True, blank=True, verbose_name=_('annotation.description'))

    text = models.TextField(null=True, blank=True, verbose_name=_('annotation.text'))

    color = models.CharField(max_length=1024, verbose_name=_('annotation.color'))

    creator = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('creator.title'))

    contributor = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('contributor.title'))

    creation_date = models.DateTimeField(auto_now_add=True, verbose_name=_('annotation.creation_date'))

    update_date = models.DateTimeField(auto_now=True, verbose_name=_('annotation.update_date'))

    @Property

    def tags(): #@NoSelf

        def fget(self):

            return ",".join(self.tag_list)

        def fset(self, value):

            values = None

            if isinstance(value, (list, tuple)):

                values = list(value)

            elif value is not None:

                values = [v.lower().strip() for v in unicode(value).split(",")]

            if values is not None:

                self.tags_field = ",".join(values) + ","

            else:

                self.tags_field = None

        return locals()

    @Property

    def tag_list(): #@NoSelf

        def fget(self):

            return [t.name for t in Tag.objects.get_for_object(self)]

        return locals()

    def get_tag_list(self):

        tags = []

        if self.tags:

            tags_list = unicode(self.tags)

            for t in tags_list.split(","):

                tags.append(t.strip()) 

        return tags

        #return self.tags

    def __unicode__(self):

        return unicode(self.external_id) + u": " + unicode(self.title)

    def serialize(self, root_element=None):

        if root_element is not None:

            iri = root_element

        else :

            iri = lxml.etree.Element('iri')

            doc = lxml.etree.ElementTree(iri)

        textannotation = lxml.etree.SubElement(iri, 'text-annotation')

        id = lxml.etree.SubElement(textannotation, 'id')

        id.text = self.external_id

        uri = lxml.etree.SubElement(textannotation, 'uri')

        uri.text = self.uri

        if self.tags:

            tags = lxml.etree.SubElement(textannotation, 'tags')

            for t in self.get_tag_list():

                tag = lxml.etree.SubElement(tags, 'tag')

                tag.text = t

        content = lxml.etree.SubElement(textannotation, 'content')

        color = lxml.etree.SubElement(content, 'color')

        color.text = self.color

        description = lxml.etree.SubElement(content, 'description')

        description.text = self.description

        title = lxml.etree.SubElement(content, 'title')

        title.text = self.title

        text = lxml.etree.SubElement(content, 'text')

        text.text = self.text

        meta = lxml.etree.SubElement(textannotation, 'meta')

        contributor = lxml.etree.SubElement(meta, "contributor")

        contributor.text = self.contributor

        creator = lxml.etree.SubElement(meta, "creator")

        creator.text = self.creator

        creationdate = lxml.etree.SubElement(meta, "created")

        creationdate.text = str(self.creation_date)

        updatedate = lxml.etree.SubElement(meta, "modified")

        updatedate.text = str(self.update_date)

        if root_element is not None:

            return root_element

        else:        

            return doc

    @staticmethod

    def create_annotation(external_id, uri=None, tags=None, title=None, description=None, text=None, color=None, creator=None, contributor=None, creation_date=None, update_date=None):

        annotation = Annotation(external_id=external_id, uri=uri, tags=tags, title=title, description=description, text=text, color=color, creator=creator, contributor=contributor, creation_date=creation_date, update_date=update_date)

        annotation.save()

        annotation.index_annot()

        return annotation

    def delete(self):

        super(Annotation, self).delete()

        lucene.getVMEnv().attachCurrentThread()

        writer = ldt.indexation.get_writer()

        try:

            writer.deleteDocuments(lucene.Term("external_id", self.external_id))

        finally:

            writer.close()

    def index_annot(self):

        lucene.getVMEnv().attachCurrentThread()

        writer = ldt.indexation.get_writer()

        try:

            annotl = [self, ]

            indexer = AnnotIndexer(annotl, writer)

            indexer.index_all()

        finally:

            writer.close()

    def update_index(self):

        lucene.getVMEnv().attachCurrentThread()

        writer = ldt.indexation.get_writer()

        try:

            writer.deleteDocuments(lucene.Term("external_id", self.external_id))

        finally:

            writer.close()

        self.index_annot()