54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
|
"""Add missing deleted_at columns to TenantMixin tables.
|
||
|
|
|
||
|
|
Several models inherit TenantMixin (which includes SoftDeleteMixin)
|
||
|
|
but their DB tables were never migrated to include the deleted_at column.
|
||
|
|
This causes 500 errors when SQLAlchemy tries to SELECT deleted_at.
|
||
|
|
|
||
|
|
Revision ID: 0083
|
||
|
|
Revises: 0082
|
||
|
|
"""
|
||
|
|
from alembic import op
|
||
|
|
import sqlalchemy as sa
|
||
|
|
|
||
|
|
revision = "0083"
|
||
|
|
down_revision = "0082"
|
||
|
|
branch_labels = None
|
||
|
|
depends_on = None
|
||
|
|
|
||
|
|
# Tables that use TenantMixin (and therefore SoftDeleteMixin) in their models
|
||
|
|
# but are missing the deleted_at column in the database.
|
||
|
|
TABLES_NEEDING_DELETED_AT = [
|
||
|
|
"contact_folder_permissions",
|
||
|
|
"permission_delegations",
|
||
|
|
"guest_users",
|
||
|
|
"entity_policies",
|
||
|
|
"notification_types",
|
||
|
|
"password_reset_tokens",
|
||
|
|
"api_tokens",
|
||
|
|
"permission_templates",
|
||
|
|
"user_groups",
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade() -> None:
|
||
|
|
conn = op.get_bind()
|
||
|
|
for table_name in TABLES_NEEDING_DELETED_AT:
|
||
|
|
# Check if column already exists before adding
|
||
|
|
result = conn.execute(sa.text(
|
||
|
|
"SELECT 1 FROM information_schema.columns "
|
||
|
|
"WHERE table_name = :t AND column_name = 'deleted_at'"
|
||
|
|
), {"t": table_name})
|
||
|
|
if result.scalar() is None:
|
||
|
|
op.add_column(
|
||
|
|
table_name,
|
||
|
|
sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True),
|
||
|
|
)
|
||
|
|
print(f" Added deleted_at to {table_name}")
|
||
|
|
else:
|
||
|
|
print(f" Skipped {table_name} (already has deleted_at)")
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade() -> None:
|
||
|
|
for table_name in reversed(TABLES_NEEDING_DELETED_AT):
|
||
|
|
op.drop_column(table_name, "deleted_at")
|