fixing migrations

This commit is contained in:
2025-12-23 20:09:47 -06:00
parent 9a0b7c7920
commit 7667d80d2f
3 changed files with 68 additions and 2 deletions

View File

@@ -119,12 +119,15 @@ def upgrade():
print(" ⚠ Table site_cidrs not found or already dropped\n")
# Step 5: Drop site_cidr_id column from site_ips
# Use batch mode for SQLite compatibility (SQLite doesn't support DROP COLUMN directly)
print("Step 5: Dropping site_cidr_id column from site_ips...")
site_ips_columns = [col['name'] for col in inspector.get_columns('site_ips')]
if 'site_cidr_id' in site_ips_columns:
try:
op.drop_column('site_ips', 'site_cidr_id')
# Use batch_alter_table for SQLite compatibility
with op.batch_alter_table('site_ips', schema=None) as batch_op:
batch_op.drop_column('site_cidr_id')
print(" ✓ Dropped site_cidr_id column from site_ips\n")
except Exception as e:
print(f" ⚠ Could not drop column: {e}\n")