Kobe/docupulse
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more robust migrations

Browse Source
This commit is contained in:
Kobe
2025-06-02 18:58:48 +02:00
parent 4dbaa27cba
commit 5c6c3f436e
51 changed files with 401 additions and 273 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
migrations/versions/dbcb5d2d3ed0_add_contact_model.py
View File

@@ -19,27 +19,30 @@ depends_on = None
def upgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.create_table('contact',
sa.Column('id', sa.Integer(), nullable=False),
sa.Column('first_name', sa.String(length=100), nullable=False),
sa.Column('last_name', sa.String(length=100), nullable=False),
sa.Column('email', sa.String(length=150), nullable=False),
sa.Column('phone', sa.String(length=20), nullable=True),
sa.Column('company', sa.String(length=100), nullable=True),
sa.Column('position', sa.String(length=100), nullable=True),
sa.Column('notes', sa.Text(), nullable=True),
sa.Column('created_at', sa.DateTime(), nullable=True),
sa.Column('updated_at', sa.DateTime(), nullable=True),
sa.Column('owner_id', sa.Integer(), nullable=False),
sa.Column('is_active', sa.Boolean(), nullable=True),
sa.ForeignKeyConstraint(['owner_id'], ['user.id'], ),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('email')
)
# Check if columns exist before adding them
conn = op.get_bind()
inspector = inspect(conn)
tables = inspector.get_table_names()
if 'contact' not in tables:
op.create_table('contact',
sa.Column('id', sa.Integer(), nullable=False),
sa.Column('first_name', sa.String(length=100), nullable=False),
sa.Column('last_name', sa.String(length=100), nullable=False),
sa.Column('email', sa.String(length=150), nullable=False),
sa.Column('phone', sa.String(length=20), nullable=True),
sa.Column('company', sa.String(length=100), nullable=True),
sa.Column('position', sa.String(length=100), nullable=True),
sa.Column('notes', sa.Text(), nullable=True),
sa.Column('created_at', sa.DateTime(), nullable=True),
sa.Column('updated_at', sa.DateTime(), nullable=True),
sa.Column('owner_id', sa.Integer(), nullable=False),
sa.Column('is_active', sa.Boolean(), nullable=True),
sa.ForeignKeyConstraint(['owner_id'], ['user.id'], ),
sa.PrimaryKeyConstraint('id'),
sa.UniqueConstraint('email')
)
# Check if columns exist before adding them
columns = [col['name'] for col in inspector.get_columns('user')]
with op.batch_alter_table('user', schema=None) as batch_op:
@@ -53,9 +56,15 @@ def upgrade():
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
conn = op.get_bind()
inspector = inspect(conn)
columns = [col['name'] for col in inspector.get_columns('user')]
with op.batch_alter_table('user', schema=None) as batch_op:
batch_op.drop_column('created_at')
batch_op.drop_column('is_admin')
if 'created_at' in columns:
batch_op.drop_column('created_at')
if 'is_admin' in columns:
batch_op.drop_column('is_admin')
op.drop_table('contact')
# ### end Alembic commands ###