48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
"""Add optional concierge association to route details.
|
|
|
|
Revision ID: 0023_detail_concierge_advisor
|
|
Revises: 0022_opaque_ids
|
|
"""
|
|
|
|
from alembic import context, op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy import inspect
|
|
from sqlalchemy import inspect
|
|
|
|
|
|
revision = "0023_detail_concierge_advisor"
|
|
down_revision = "0022_opaque_ids"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def should_add_concierge_advisor_column(existing_columns: set[str]) -> bool:
|
|
return "conciergeAdvisorId" not in existing_columns
|
|
|
|
|
|
def upgrade() -> None:
|
|
if context.is_offline_mode():
|
|
should_add = True
|
|
else:
|
|
existing_columns = {
|
|
column["name"]
|
|
for column in inspect(op.get_bind()).get_columns("DetailRecord")
|
|
}
|
|
should_add = should_add_concierge_advisor_column(existing_columns)
|
|
|
|
if should_add:
|
|
op.add_column("DetailRecord", sa.Column("conciergeAdvisorId", sa.String(), nullable=True))
|
|
|
|
|
|
def downgrade() -> None:
|
|
if context.is_offline_mode():
|
|
op.drop_column("DetailRecord", "conciergeAdvisorId")
|
|
return
|
|
|
|
existing_columns = {
|
|
column["name"]
|
|
for column in inspect(op.get_bind()).get_columns("DetailRecord")
|
|
}
|
|
if "conciergeAdvisorId" in existing_columns:
|
|
op.drop_column("DetailRecord", "conciergeAdvisorId")
|