39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
|
"""Add compliance metadata columns to ai_providers table (B-AIPROV-COMP).
|
||
|
|
|
||
|
|
Adds region, hosting_type, dpa_status, retention_policy,
|
||
|
|
training_on_customer_data, transfer_notice, allowed_data_classes
|
||
|
|
to support AI provider compliance checks.
|
||
|
|
|
||
|
|
Revision ID: 0119
|
||
|
|
Revises: 0118
|
||
|
|
"""
|
||
|
|
|
||
|
|
from alembic import op
|
||
|
|
import sqlalchemy as sa
|
||
|
|
from sqlalchemy.dialects.postgresql import JSONB
|
||
|
|
|
||
|
|
revision = "0119"
|
||
|
|
down_revision = "0118"
|
||
|
|
branch_labels = None
|
||
|
|
depends_on = None
|
||
|
|
|
||
|
|
|
||
|
|
def upgrade() -> None:
|
||
|
|
op.add_column("ai_providers", sa.Column("region", sa.String(20), nullable=False, server_default="unknown"))
|
||
|
|
op.add_column("ai_providers", sa.Column("hosting_type", sa.String(30), nullable=False, server_default="cloud"))
|
||
|
|
op.add_column("ai_providers", sa.Column("dpa_status", sa.String(20), nullable=False, server_default="none"))
|
||
|
|
op.add_column("ai_providers", sa.Column("retention_policy", sa.Text(), nullable=False, server_default=""))
|
||
|
|
op.add_column("ai_providers", sa.Column("training_on_customer_data", sa.Boolean(), nullable=False, server_default=sa.text("false")))
|
||
|
|
op.add_column("ai_providers", sa.Column("transfer_notice", sa.Text(), nullable=False, server_default=""))
|
||
|
|
op.add_column("ai_providers", sa.Column("allowed_data_classes", JSONB, nullable=False, server_default=sa.text("'[]'::jsonb")))
|
||
|
|
|
||
|
|
|
||
|
|
def downgrade() -> None:
|
||
|
|
op.drop_column("ai_providers", "allowed_data_classes")
|
||
|
|
op.drop_column("ai_providers", "transfer_notice")
|
||
|
|
op.drop_column("ai_providers", "training_on_customer_data")
|
||
|
|
op.drop_column("ai_providers", "retention_policy")
|
||
|
|
op.drop_column("ai_providers", "dpa_status")
|
||
|
|
op.drop_column("ai_providers", "hosting_type")
|
||
|
|
op.drop_column("ai_providers", "region")
|