Files
arrr-erp/alembic/versions/20260530_phase_7q2_firm_branding.py
2026-06-20 15:01:44 +05:30

56 lines
2.4 KiB
Python

"""Phase 7Q.2 firm branding settings
Revision ID: 20260530_phase_7q2_firm_branding
Revises: 20260529_phase_7h_common_alerts
Create Date: 2026-05-21
Adds tenant-level branding fields and branch-level invoice/payment presentation fields.
The migration is idempotent for SQLite/dev environments.
"""
from __future__ import annotations
from alembic import op
import sqlalchemy as sa
revision = "20260530_phase_7q2_firm_branding"
down_revision = "20260529_phase_7h_common_alerts"
branch_labels = None
depends_on = None
def _columns(table_name: str) -> set[str]:
bind = op.get_bind()
inspector = sa.inspect(bind)
if table_name not in inspector.get_table_names():
return set()
return {c["name"] for c in inspector.get_columns(table_name)}
def _add_if_missing(table: str, column: sa.Column) -> None:
if column.name not in _columns(table):
op.add_column(table, column)
def upgrade() -> None:
_add_if_missing("tenants", sa.Column("display_name", sa.String(length=200), nullable=True))
_add_if_missing("tenants", sa.Column("logo_path", sa.String(length=500), nullable=True))
_add_if_missing("tenants", sa.Column("favicon_path", sa.String(length=500), nullable=True))
_add_if_missing("tenants", sa.Column("primary_color", sa.String(length=20), nullable=True))
_add_if_missing("tenants", sa.Column("accent_color", sa.String(length=20), nullable=True))
_add_if_missing("tenants", sa.Column("website_url", sa.String(length=255), nullable=True))
_add_if_missing("tenants", sa.Column("contact_email", sa.String(length=255), nullable=True))
_add_if_missing("tenants", sa.Column("contact_mobile", sa.String(length=50), nullable=True))
_add_if_missing("branch_settings", sa.Column("invoice_footer_text", sa.Text(), nullable=True))
_add_if_missing("branch_settings", sa.Column("bank_name", sa.String(length=200), nullable=True))
_add_if_missing("branch_settings", sa.Column("bank_account_name", sa.String(length=200), nullable=True))
_add_if_missing("branch_settings", sa.Column("bank_account_number", sa.String(length=50), nullable=True))
_add_if_missing("branch_settings", sa.Column("bank_ifsc", sa.String(length=20), nullable=True))
_add_if_missing("branch_settings", sa.Column("upi_id", sa.String(length=100), nullable=True))
def downgrade() -> None:
# SQLite cannot reliably drop columns on older setups without table rebuild.
# Keep downgrade no-op to avoid damaging live tenant/branch settings.
pass