add fields to media

2 files changed, 59 insertions, 0 deletions

diff --git a/animism-align/cli/app/sql/models/media.py b/animism-align/cli/app/sql/models/media.py
index 6b615c9..27a1498 100644
--- a/animism-align/cli/app/sql/models/media.py
+++ b/animism-align/cli/app/sql/models/media.py
@@ -13,12 +13,28 @@ class Media(Base):
   __tablename__ = 'media'
   id = Column(Integer, primary_key=True)
   type = Column(String(16, 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)
+  pre_title = Column(String(256, convert_unicode=True), nullable=True)
+  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)
+  start_ts = Column(Float, nullable=True)
   settings = Column(JSON, default={}, nullable=True)
 
   def toJSON(self):
     return {
       'id': self.id,
       'type': self.type,
+      'url': self.url,
+      'title': self.title,
+      'pre_title': self.pre_title,
+      'translated_title': self.translated_title,
+      'date': self.date,
+      'source': self.source,
+      'author': self.author,
+      'start_ts': self.start_ts,
       'settings': self.settings,
     }
 
diff --git a/animism-align/cli/app/sql/versions/202007080030_add_fields_to_media.py b/animism-align/cli/app/sql/versions/202007080030_add_fields_to_media.py
new file mode 100644
index 0000000..35d0e14
--- /dev/null
+++ b/animism-align/cli/app/sql/versions/202007080030_add_fields_to_media.py
@@ -0,0 +1,43 @@
+"""add fields to media
+
+Revision ID: c93b3c61b053
+Revises: d319ace44bec
+Create Date: 2020-07-08 00:30:30.280537
+
+"""
+from alembic import op
+import sqlalchemy as sa
+import sqlalchemy_utc
+
+
+# revision identifiers, used by Alembic.
+revision = 'c93b3c61b053'
+down_revision = 'd319ace44bec'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('media', sa.Column('author', sa.String(length=256, _expect_unicode=True), nullable=True))
+    op.add_column('media', sa.Column('date', sa.String(length=256, _expect_unicode=True), nullable=True))
+    op.add_column('media', sa.Column('pre_title', sa.String(length=256, _expect_unicode=True), nullable=True))
+    op.add_column('media', sa.Column('source', sa.String(length=256, _expect_unicode=True), nullable=True))
+    op.add_column('media', sa.Column('start_ts', sa.Float(), nullable=True))
+    op.add_column('media', sa.Column('title', sa.String(length=256, _expect_unicode=True), nullable=True))
+    op.add_column('media', sa.Column('translated_title', sa.String(length=256, _expect_unicode=True), nullable=True))
+    op.add_column('media', sa.Column('url', sa.String(length=256, _expect_unicode=True), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('media', 'url')
+    op.drop_column('media', 'translated_title')
+    op.drop_column('media', 'title')
+    op.drop_column('media', 'start_ts')
+    op.drop_column('media', 'source')
+    op.drop_column('media', 'pre_title')
+    op.drop_column('media', 'date')
+    op.drop_column('media', 'author')
+    # ### end Alembic commands ###