1 files changed, 39 insertions, 0 deletions
diff --git a/animism-align/cli/app/sql/models/annotation.py b/animism-align/cli/app/sql/models/annotation.py
new file mode 100644
index 0000000..6cc476c
--- /dev/null
+++ b/animism-align/cli/app/sql/models/annotation.py
@@ -0,0 +1,39 @@
+from sqlalchemy import create_engine, Table, Column, Text, String, Integer, Float, DateTime, JSON, ForeignKey
+from sqlalchemy.orm import relationship
+import sqlalchemy.sql.functions as func
+from sqlalchemy_utc import UtcDateTime, utcnow
+from wtforms_alchemy import ModelForm
+
+from app.sql.common import db, Base, Session
+# from app.sql.models.page import Page
+
+from app.settings import app_cfg
+
+class Annotation(Base):
+  """Table for storing references to graphs"""
+  __tablename__ = 'annotation'
+  id = Column(Integer, primary_key=True)
+  type = Column(String(16, convert_unicode=True), nullable=False)
+  paragraph_id = Column(Integer, ForeignKey('paragraph.id'), nullable=True)
+  start_ts = Column(Float, nullable=False)
+  end_ts = Column(Float, nullable=True)
+  text = Column(Text(convert_unicode=True), nullable=True)
+  settings = Column(JSON, default={}, nullable=True)
+
+  def toJSON(self):
+    return {
+      'id': self.id,
+      'type': self.type,
+      'paragraph_id': self.paragraph_id,
+      'start_ts': self.start_ts,
+      'end_ts': self.end_ts,
+      'sentence': self.description,
+      'settings': self.settings,
+    }
+
+class AnnotationForm(ModelForm):
+  class Meta:
+    model = Annotation
+    exclude = ['settings']
+  def get_session():
+    return Session()