web/ldt/text/models.py
changeset 17 683ce4109c28
parent 9 22ab430e9b64
child 21 1a061f244254
child 22 83b28fc0d731
--- a/web/ldt/text/models.py	Wed Nov 17 18:57:34 2010 +0100
+++ b/web/ldt/text/models.py	Fri Nov 19 14:29:17 2010 +0100
@@ -4,16 +4,21 @@
 from ldt.core.models import Document, Owner
 from django.contrib.auth.models import User
 import tagging.fields
+from tagging.models import Tag
 from utils import generate_uuid
 import os.path
 import uuid
+import lxml
 #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 = models.CharField(max_length=1024, null=True, blank=True, verbose_name=_('annotation.tags'))
+    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'))
@@ -22,16 +27,99 @@
     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():
+        
+        def fget(self):
+            return ",".join(self.tag_list)
+        
+        def fset(self, value):
+            values = None
+            if type(value) == type([]):
+                values = 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():
+        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()
         
         return annotation
-