add project type

1 files changed, 2 insertions, 0 deletions

diff --git a/animism-align/cli/app/sql/models/media.py b/animism-align/cli/app/sql/models/media.py
index 4628f4d..ae16255 100644
--- a/animism-align/cli/app/sql/models/media.py
+++ b/animism-align/cli/app/sql/models/media.py
@@ -12,6 +12,7 @@ class Media(Base):
   """Table for storing references to media"""
   __tablename__ = 'media'
   id = Column(Integer, primary_key=True)
+  episode_id = Column(Integer)
   type = Column(String(16, convert_unicode=True), nullable=False)
   tag = Column(String(64, convert_unicode=True), nullable=True)
   url = Column(String(256, convert_unicode=True), nullable=True)
@@ -30,6 +31,7 @@ class Media(Base):
   def toJSON(self):
     return {
       'id': self.id,
+      'episode_id': self.episode_id,
       'type': self.type,
       'tag': self.tag,
       'url': self.url,