fixed problematic migration file

Author: ludavidca

backend/python/migrations/versions/ba76119b3e4c_update_user_system.py

@@ -19,47 +19,85 @@
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
-    op.create_table('system_settings',
-    sa.Column('created_at', sa.DateTime(), nullable=True),
-    sa.Column('updated_at', sa.DateTime(), nullable=True),
-    sa.Column('default_cap', sa.Integer(), nullable=True),
-    sa.Column('route_start_time', sa.Time(), nullable=True),
-    sa.Column('warehouse_location', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
-    sa.Column('warehouse_longitude', sa.Float(), nullable=True),
-    sa.Column('warehouse_latitude', sa.Float(), nullable=True),
-    sa.Column('system_settings_id', sa.Uuid(), nullable=False),
-    sa.PrimaryKeyConstraint('system_settings_id')
-    )
-    op.create_table('users',
-    sa.Column('created_at', sa.DateTime(), nullable=True),
-    sa.Column('updated_at', sa.DateTime(), nullable=True),
-    sa.Column('name', sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False),
-    sa.Column('email', sqlmodel.sql.sqltypes.AutoString(length=254), nullable=False),
-    sa.Column('user_id', sa.Uuid(), nullable=False),
-    sa.Column('auth_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
-    sa.Column('role', sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False),
-    sa.PrimaryKeyConstraint('user_id')
-    )
-    op.create_index(op.f('ix_users_auth_id'), 'users', ['auth_id'], unique=True)
-    op.create_index(op.f('ix_users_email'), 'users', ['email'], unique=True)
-    op.create_index(op.f('ix_users_user_id'), 'users', ['user_id'], unique=False)
-    op.add_column('admin_info', sa.Column('receive_email_notifications', sa.Boolean(), nullable=False))
-    op.add_column('admin_info', sa.Column('user_id', sa.Uuid(), nullable=False))
+    # Check if table exists before creating
+    from sqlalchemy import inspect
+    bind = op.get_bind()
+    inspector = inspect(bind)
+    existing_tables = inspector.get_table_names()
+
+    if 'system_settings' not in existing_tables:
+        op.create_table('system_settings',
+        sa.Column('created_at', sa.DateTime(), nullable=True),
+        sa.Column('updated_at', sa.DateTime(), nullable=True),
+        sa.Column('default_cap', sa.Integer(), nullable=True),
+        sa.Column('route_start_time', sa.Time(), nullable=True),
+        sa.Column('warehouse_location', sqlmodel.sql.sqltypes.AutoString(), nullable=True),
+        sa.Column('warehouse_longitude', sa.Float(), nullable=True),
+        sa.Column('warehouse_latitude', sa.Float(), nullable=True),
+        sa.Column('system_settings_id', sa.Uuid(), nullable=False),
+        sa.PrimaryKeyConstraint('system_settings_id')
+        )
+
+    if 'users' not in existing_tables:
+        op.create_table('users',
+        sa.Column('created_at', sa.DateTime(), nullable=True),
+        sa.Column('updated_at', sa.DateTime(), nullable=True),
+        sa.Column('name', sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False),
+        sa.Column('email', sqlmodel.sql.sqltypes.AutoString(length=254), nullable=False),
+        sa.Column('user_id', sa.Uuid(), nullable=False),
+        sa.Column('auth_id', sqlmodel.sql.sqltypes.AutoString(), nullable=False),
+        sa.Column('role', sqlmodel.sql.sqltypes.AutoString(length=255), nullable=False),
+        sa.PrimaryKeyConstraint('user_id')
+        )
+        op.create_index(op.f('ix_users_auth_id'), 'users', ['auth_id'], unique=True)
+        op.create_index(op.f('ix_users_email'), 'users', ['email'], unique=True)
+        op.create_index(op.f('ix_users_user_id'), 'users', ['user_id'], unique=False)
+
+    # Check existing columns in admin_info
+    admin_info_columns = [col['name'] for col in inspector.get_columns('admin_info')]
+
+    if 'receive_email_notifications' not in admin_info_columns:
+        op.add_column('admin_info', sa.Column('receive_email_notifications', sa.Boolean(), nullable=False))
+    if 'user_id' not in admin_info_columns:
+        op.add_column('admin_info', sa.Column('user_id', sa.Uuid(), nullable=False))
     op.create_unique_constraint(None, 'admin_info', ['user_id'])
     op.create_foreign_key(None, 'admin_info', 'users', ['user_id'], ['user_id'])
-    op.drop_column('admin_info', 'admin_name')
-    op.drop_column('admin_info', 'default_cap')
-    op.drop_column('admin_info', 'route_start_time')
-    op.drop_column('admin_info', 'warehouse_location')
-    op.drop_column('admin_info', 'admin_email')
-    op.add_column('drivers', sa.Column('user_id', sa.Uuid(), nullable=False))
-    op.drop_index(op.f('ix_drivers_auth_id'), table_name='drivers')
-    op.drop_index(op.f('ix_drivers_email'), table_name='drivers')
+
+    # Drop columns only if they exist
+    if 'admin_name' in admin_info_columns:
+        op.drop_column('admin_info', 'admin_name')
+    if 'default_cap' in admin_info_columns:
+        op.drop_column('admin_info', 'default_cap')
+    if 'route_start_time' in admin_info_columns:
+        op.drop_column('admin_info', 'route_start_time')
+    if 'warehouse_location' in admin_info_columns:
+        op.drop_column('admin_info', 'warehouse_location')
+    if 'admin_email' in admin_info_columns:
+        op.drop_column('admin_info', 'admin_email')
+
+    # Check existing columns in drivers
+    drivers_columns = [col['name'] for col in inspector.get_columns('drivers')]
+
+    if 'user_id' not in drivers_columns:
+        op.add_column('drivers', sa.Column('user_id', sa.Uuid(), nullable=False))
+
+    # Drop indexes only if they exist
+    existing_indexes = [idx['name'] for idx in inspector.get_indexes('drivers')]
+    if 'ix_drivers_auth_id' in existing_indexes:
+        op.drop_index(op.f('ix_drivers_auth_id'), table_name='drivers')
+    if 'ix_drivers_email' in existing_indexes:
+        op.drop_index(op.f('ix_drivers_email'), table_name='drivers')
+
     op.create_unique_constraint(None, 'drivers', ['user_id'])
     op.create_foreign_key(None, 'drivers', 'users', ['user_id'], ['user_id'])
-    op.drop_column('drivers', 'name')
-    op.drop_column('drivers', 'email')
-    op.drop_column('drivers', 'auth_id')
+
+    # Drop columns only if they exist
+    if 'name' in drivers_columns:
+        op.drop_column('drivers', 'name')
+    if 'email' in drivers_columns:
+        op.drop_column('drivers', 'email')
+    if 'auth_id' in drivers_columns:
+        op.drop_column('drivers', 'auth_id')
     # ### end Alembic commands ###