add fields to media part 2

2 files changed, 6 insertions, 0 deletions

diff --git a/animism-align/cli/app/sql/models/media.py b/animism-align/cli/app/sql/models/media.py
index 27a1498..77b99ca 100644
--- a/animism-align/cli/app/sql/models/media.py
+++ b/animism-align/cli/app/sql/models/media.py
@@ -13,6 +13,7 @@ class Media(Base):
   __tablename__ = 'media'
   id = Column(Integer, primary_key=True)
   type = Column(String(16, convert_unicode=True), nullable=False)
+  tag = Column(String(64, convert_unicode=True), nullable=False)
   url = Column(String(256, convert_unicode=True), nullable=True)
   title = Column(String(256, convert_unicode=True), nullable=True)
   author = Column(String(256, convert_unicode=True), nullable=True)
@@ -20,6 +21,7 @@ class Media(Base):
   translated_title = Column(String(256, convert_unicode=True), nullable=True)
   date = Column(String(256, convert_unicode=True), nullable=True)
   source = Column(String(256, convert_unicode=True), nullable=True)
+  medium = Column(String(64, convert_unicode=True), nullable=True)
   start_ts = Column(Float, nullable=True)
   settings = Column(JSON, default={}, nullable=True)
 
@@ -27,6 +29,7 @@ class Media(Base):
     return {
       'id': self.id,
       'type': self.type,
+      'tag': self.tag,
       'url': self.url,
       'title': self.title,
       'pre_title': self.pre_title,
@@ -34,6 +37,7 @@ class Media(Base):
       'date': self.date,
       'source': self.source,
       'author': self.author,
+      'medium': self.medium,
       'start_ts': self.start_ts,
       'settings': self.settings,
     }
diff --git a/animism-align/cli/app/sql/models/upload.py b/animism-align/cli/app/sql/models/upload.py
index 8680f19..7e8a56e 100644
--- a/animism-align/cli/app/sql/models/upload.py
+++ b/animism-align/cli/app/sql/models/upload.py
@@ -17,6 +17,7 @@ class Upload(Base):
   sha256 = Column(String(256), nullable=False)
   fn = Column(String(256), nullable=False)
   ext = Column(String(4, convert_unicode=True), nullable=False)
+  tag = Column(String(64, convert_unicode=True), nullable=False)
   username = Column(String(16, convert_unicode=True), nullable=False)
   created_at = Column(UtcDateTime(), default=utcnow())
 
@@ -26,6 +27,7 @@ class Upload(Base):
       'sha256': self.sha256,
       'fn': self.fn,
       'ext': self.ext,
+      'tag': self.tag,
       'username': self.username,
       'url': self.url(),
       'created_at': self.created_at,