hide peers

oml/annotation/models.py

@@ -0,0 +1,78 @@
+# -*- coding: utf-8 -*-
+from datetime import datetime
+import json
+import logging
+import os
+import shutil
+
+from sqlalchemy.orm import load_only
+import ox
+import sqlalchemy as sa
+
+from changelog import add_record
+from db import MutableDict
+import db
+import json_pickler
+import settings
+import state
+import utils
+import media
+from websocket import trigger_event
+
+
+logger = logging.getLogger(__name__)
+
+
+class Annotation(db.Model):
+    __tablename__ = 'annotation'
+
+    _id = sa.Column(sa.Integer(), primary_key=True)
+    id = sa.Column(sa.String(43))
+    created = sa.Column(sa.DateTime())
+    modified = sa.Column(sa.DateTime())
+
+    user = sa.orm.relationship('User', backref=sa.orm.backref('annotations', lazy='dynamic'))
+    item = sa.orm.relationship('Item', backref=sa.orm.backref('items', lazy='dynamic'))
+    data = sa.Column(MutableDict.as_mutable(sa.PickleType(pickler=json_pickler)))
+
+    def __init__(self, item_id, user_id, data):
+        self.created = datetime.utcnow()
+        self.modified = datetime.utcnow()
+        self.item_id = item_id
+        self.user_id = user_id
+        self.data = data
+
+    @classmethod
+    def create(cls, **kwargs):
+        a = cls(**kwargs)
+        state.db.session.add(a)
+        state.db.session.commit()
+
+    @classmethod
+    def get(cls, user, item_id, annotation_id):
+        for a in cls.query.filter_by(item_id=item_id, user=user, _id=annotation_id):
+            if a.data.get('id') == annotation_id:
+                return a
+
+    @classmethod
+    def get_by_item(cls, user, item_id):
+        annotations = []
+        for a in cls.query.filter_by(item_id=item_id):
+            annotations.append(a.json())
+        return annotations
+
+    def save(self):
+        _id = self.data.get('id')
+        if _id:
+            self._id = _id
+        state.db.session.add(self)
+        state.db.session.commit()
+
+    def json(self):
+        data = self.data.copy()
+        data['created'] = self.created
+        data['modified'] = self.modified
+        data['user'] = self.user_id
+        data['_id'] = ox.toAZ(self.id)
+        return data
+