From 00c429034437ae2f5540d073d79b3c4bba92c547 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 8 Jul 2020 10:33:48 +0200
Subject: add fields to media part 2

---
 animism-align/cli/app/sql/models/media.py  | 4 ++++
 animism-align/cli/app/sql/models/upload.py | 2 ++
 2 files changed, 6 insertions(+)

(limited to 'animism-align')

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,
-- 
cgit v1.2.3-70-g09d2