From 462f6d3b21558ed0a283c24e0e0332eac6ccbbb3 Mon Sep 17 00:00:00 2001 From: François Schmidts Date: Fri, 11 Sep 2015 18:28:12 +0200 Subject: base modification in model for category support --- migrations/versions/3f83bfe93fc_adding_category.py | 42 ++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 migrations/versions/3f83bfe93fc_adding_category.py (limited to 'migrations') diff --git a/migrations/versions/3f83bfe93fc_adding_category.py b/migrations/versions/3f83bfe93fc_adding_category.py new file mode 100644 index 00000000..bd65a01f --- /dev/null +++ b/migrations/versions/3f83bfe93fc_adding_category.py @@ -0,0 +1,42 @@ +"""adding category + +Revision ID: 3f83bfe93fc +Revises: 25ca960a207 +Create Date: 2015-09-01 14:15:04.212563 +""" + +# revision identifiers, used by Alembic. +revision = '3f83bfe93fc' +down_revision = '25ca960a207' + +import conf +from alembic import op +import sqlalchemy as sa + + +def upgrade(): + op.create_table('category', + sa.Column('id', sa.Integer(), nullable=False), + sa.Column('name', sa.String(), nullable=True), + sa.Column('user_id', sa.Integer(), nullable=True), + sa.ForeignKeyConstraint(['user_id'], ['user.id'], ), + sa.PrimaryKeyConstraint('id')) + op.add_column('article', + sa.Column('category_id', sa.Integer(), nullable=True)) + op.add_column('feed', + sa.Column('category_id', sa.Integer(), nullable=True)) + if 'sqlite' not in conf.SQLALCHEMY_DATABASE_URI: + op.create_foreign_key(None, 'article', 'category', + ['category_id'], ['id']) + op.create_foreign_key(None, 'feed', 'category', + ['category_id'], ['id']) + + +def downgrade(): + if 'sqlite' not in conf.SQLALCHEMY_DATABASE_URI: + op.drop_constraint(None, 'feed', type_='foreignkey') + op.drop_constraint(None, 'feed', type_='foreignkey') + op.drop_column('feed', 'category_id') + op.drop_constraint(None, 'article', type_='foreignkey') + op.drop_column('article', 'category_id') + op.drop_table('category') -- cgit