A R R R Associates
81 lines
3.1 KiB
Python
81 lines
3.1 KiB
Python
"""Phase 11B client-aware Bank Analyzer ownership and engagement archive.
|
|
|
|
Revision ID: 20260822_bank_client_archive_p11b
|
|
Revises: 20260822_bank_to_tally_p11
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
|
|
revision = "20260822_bank_client_archive_p11b"
|
|
down_revision = "20260822_bank_to_tally_p11"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
op.add_column(
|
|
"bank_statement_analysis_jobs",
|
|
sa.Column("client_id", sa.Integer(), sa.ForeignKey("clients.id", ondelete="SET NULL"), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"bank_statement_analysis_jobs",
|
|
sa.Column("engagement_id", sa.Integer(), sa.ForeignKey("client_service_subscriptions.id", ondelete="SET NULL"), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"bank_statement_analysis_jobs",
|
|
sa.Column("ownership_confirmation", sa.Boolean(), nullable=False, server_default=sa.false()),
|
|
)
|
|
op.add_column(
|
|
"bank_statement_analysis_jobs",
|
|
sa.Column("ownership_status", sa.String(30), nullable=False, server_default="not_checked"),
|
|
)
|
|
op.add_column(
|
|
"bank_statement_analysis_jobs",
|
|
sa.Column("ownership_validation_json", sa.Text(), nullable=True),
|
|
)
|
|
op.add_column(
|
|
"bank_statement_analysis_jobs",
|
|
sa.Column("engagement_archive_status", sa.String(30), nullable=False, server_default="not_requested"),
|
|
)
|
|
op.add_column(
|
|
"bank_statement_analysis_jobs",
|
|
sa.Column("engagement_archive_json", sa.Text(), nullable=True),
|
|
)
|
|
|
|
op.create_index(
|
|
"ix_bank_statement_analysis_jobs_client_id",
|
|
"bank_statement_analysis_jobs",
|
|
["client_id"],
|
|
)
|
|
op.create_index(
|
|
"ix_bank_statement_analysis_jobs_engagement_id",
|
|
"bank_statement_analysis_jobs",
|
|
["engagement_id"],
|
|
)
|
|
op.create_index(
|
|
"ix_bank_statement_analysis_jobs_ownership_status",
|
|
"bank_statement_analysis_jobs",
|
|
["ownership_status"],
|
|
)
|
|
op.create_index(
|
|
"ix_bank_statement_analysis_jobs_engagement_archive_status",
|
|
"bank_statement_analysis_jobs",
|
|
["engagement_archive_status"],
|
|
)
|
|
|
|
|
|
def downgrade():
|
|
op.drop_index("ix_bank_statement_analysis_jobs_engagement_archive_status", table_name="bank_statement_analysis_jobs")
|
|
op.drop_index("ix_bank_statement_analysis_jobs_ownership_status", table_name="bank_statement_analysis_jobs")
|
|
op.drop_index("ix_bank_statement_analysis_jobs_engagement_id", table_name="bank_statement_analysis_jobs")
|
|
op.drop_index("ix_bank_statement_analysis_jobs_client_id", table_name="bank_statement_analysis_jobs")
|
|
|
|
op.drop_column("bank_statement_analysis_jobs", "engagement_archive_json")
|
|
op.drop_column("bank_statement_analysis_jobs", "engagement_archive_status")
|
|
op.drop_column("bank_statement_analysis_jobs", "ownership_validation_json")
|
|
op.drop_column("bank_statement_analysis_jobs", "ownership_status")
|
|
op.drop_column("bank_statement_analysis_jobs", "ownership_confirmation")
|
|
op.drop_column("bank_statement_analysis_jobs", "engagement_id")
|
|
op.drop_column("bank_statement_analysis_jobs", "client_id")
|