A R R R Associates
81 lines
4.6 KiB
Python
81 lines
4.6 KiB
Python
"""Phase 3 client business intelligence profile foundation.
|
|
|
|
Revision ID: 20260822_business_profile_p3
|
|
Revises: 20260822_lean_agent_p2
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "20260822_business_profile_p3"
|
|
down_revision = "20260822_lean_agent_p2"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
op.create_table(
|
|
"client_business_profiles",
|
|
sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column("client_id", sa.Integer(), sa.ForeignKey("clients.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("tenant_id", sa.Integer(), sa.ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("branch_id", sa.Integer(), sa.ForeignKey("branches.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("primary_industry", sa.String(length=150), nullable=True),
|
|
sa.Column("primary_business_activity", sa.String(length=200), nullable=True),
|
|
sa.Column("secondary_business_activities", sa.Text(), nullable=True),
|
|
sa.Column("business_model", sa.String(length=50), nullable=True),
|
|
sa.Column("main_products", sa.Text(), nullable=True),
|
|
sa.Column("main_services", sa.Text(), nullable=True),
|
|
sa.Column("inventory_maintained", sa.Boolean(), nullable=True),
|
|
sa.Column("project_job_based", sa.Boolean(), nullable=True),
|
|
sa.Column("capital_intensive", sa.Boolean(), nullable=True),
|
|
sa.Column("vehicle_intensive", sa.Boolean(), nullable=True),
|
|
sa.Column("profile_status", sa.String(length=30), nullable=False, server_default="draft"),
|
|
sa.Column("confidence_score", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("confirmed_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("confirmed_at_utc", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("notes", sa.Text(), nullable=True),
|
|
sa.Column("created_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("updated_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at_utc", sa.DateTime(timezone=True), nullable=False),
|
|
sa.UniqueConstraint("client_id", name="uq_client_business_profiles_client"),
|
|
)
|
|
for name, cols in [
|
|
("ix_client_business_profiles_client_id", ["client_id"]),
|
|
("ix_client_business_profiles_tenant_id", ["tenant_id"]),
|
|
("ix_client_business_profiles_branch_id", ["branch_id"]),
|
|
("ix_client_business_profiles_primary_industry", ["primary_industry"]),
|
|
("ix_client_business_profiles_business_model", ["business_model"]),
|
|
("ix_client_business_profiles_profile_status", ["profile_status"]),
|
|
]:
|
|
op.create_index(name, "client_business_profiles", cols)
|
|
|
|
op.create_table(
|
|
"client_business_profile_sources",
|
|
sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column("profile_id", sa.Integer(), sa.ForeignKey("client_business_profiles.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("client_id", sa.Integer(), sa.ForeignKey("clients.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("tenant_id", sa.Integer(), sa.ForeignKey("tenants.id", ondelete="CASCADE"), nullable=False),
|
|
sa.Column("source_type", sa.String(length=40), nullable=False),
|
|
sa.Column("source_reference", sa.String(length=255), nullable=True),
|
|
sa.Column("observed_on", sa.Date(), nullable=True),
|
|
sa.Column("confidence_score", sa.Integer(), nullable=False, server_default="50"),
|
|
sa.Column("is_authoritative", sa.Boolean(), nullable=False, server_default=sa.false()),
|
|
sa.Column("evidence_json", sa.JSON(), nullable=True),
|
|
sa.Column("notes", sa.Text(), nullable=True),
|
|
sa.Column("created_by_user_id", sa.Integer(), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False),
|
|
)
|
|
for name, cols in [
|
|
("ix_client_business_profile_sources_profile_id", ["profile_id"]),
|
|
("ix_client_business_profile_sources_client_id", ["client_id"]),
|
|
("ix_client_business_profile_sources_tenant_id", ["tenant_id"]),
|
|
("ix_client_business_profile_sources_source_type", ["source_type"]),
|
|
]:
|
|
op.create_index(name, "client_business_profile_sources", cols)
|
|
|
|
|
|
def downgrade():
|
|
op.drop_table("client_business_profile_sources")
|
|
op.drop_table("client_business_profiles")
|