diff options
| -rw-r--r-- | alembic/versions/b1fdfe71fcb7_channel_category_foreign_key.py | 28 | ||||
| -rw-r--r-- | metricity/models.py | 6 |
2 files changed, 33 insertions, 1 deletions
diff --git a/alembic/versions/b1fdfe71fcb7_channel_category_foreign_key.py b/alembic/versions/b1fdfe71fcb7_channel_category_foreign_key.py new file mode 100644 index 0000000..329924b --- /dev/null +++ b/alembic/versions/b1fdfe71fcb7_channel_category_foreign_key.py @@ -0,0 +1,28 @@ +"""channel category foreign key + +Revision ID: b1fdfe71fcb7 +Revises: 38085c8f1099 +Create Date: 2020-08-25 15:00:38.666504 + +""" +from alembic import op +import sqlalchemy as sa + + +# revision identifiers, used by Alembic. +revision = 'b1fdfe71fcb7' +down_revision = '38085c8f1099' +branch_labels = None +depends_on = None + + +def upgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.create_foreign_key(None, 'channels', 'categories', ['category_id'], ['id']) + # ### end Alembic commands ### + + +def downgrade(): + # ### commands auto generated by Alembic - please adjust! ### + op.drop_constraint(None, 'channels', type_='foreignkey') + # ### end Alembic commands ### diff --git a/metricity/models.py b/metricity/models.py index d5cc663..43e14d3 100644 --- a/metricity/models.py +++ b/metricity/models.py @@ -24,7 +24,11 @@ class Channel(db.Model): id = db.Column(db.BigInteger, primary_key=True) name = db.Column(db.String, nullable=False) - category_id = db.Column(db.BigInteger, nullable=True) + category_id = db.Column( + db.BigInteger, + db.ForeignKey("categories.id"), + nullable=True + ) is_staff = db.Column(db.Boolean, nullable=False) |