"""add profiles points ledger sessions messages schema Revision ID: 202604030002 Revises: 202604020001 Create Date: 2026-04-03 22:40:00 """ from typing import Sequence, Union from alembic import op import sqlalchemy as sa from sqlalchemy.dialects import postgresql revision: str = "202604030002" down_revision: Union[str, Sequence[str], None] = "202604020001" branch_labels: Union[str, Sequence[str], None] = None depends_on: Union[str, Sequence[str], None] = None def upgrade() -> None: op.create_table( "profiles", sa.Column("id", sa.UUID(), nullable=False), sa.Column("username", sa.String(length=30), nullable=False), sa.Column("avatar_url", sa.Text(), nullable=True), sa.Column("bio", sa.String(length=200), nullable=True), sa.Column( "settings", postgresql.JSONB(astext_type=sa.Text()), nullable=False, server_default=sa.text("'{}'::jsonb"), ), sa.Column( "created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column( "updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True), sa.CheckConstraint( "char_length(username) >= 1", name="ck_profiles_username_non_empty" ), sa.ForeignKeyConstraint(["id"], ["auth.users.id"], ondelete="CASCADE"), sa.PrimaryKeyConstraint("id"), ) op.create_index("ix_profiles_username", "profiles", ["username"], unique=False) op.create_index( "ix_profiles_settings_gin", "profiles", ["settings"], unique=False, postgresql_using="gin", ) _enable_rls("profiles") op.create_table( "sessions", sa.Column("id", sa.UUID(), nullable=False), sa.Column("user_id", sa.UUID(), nullable=False), sa.Column("session_type", sa.String(length=20), nullable=False), sa.Column("job_id", sa.UUID(), nullable=True), sa.Column("title", sa.String(length=255), nullable=True), sa.Column("status", sa.String(length=20), nullable=False), sa.Column( "last_activity_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column( "message_count", sa.Integer(), server_default=sa.text("0"), nullable=False, ), sa.Column( "total_tokens", sa.Integer(), server_default=sa.text("0"), nullable=False, ), sa.Column( "total_cost", sa.Numeric(12, 6), server_default=sa.text("0"), nullable=False, ), sa.Column( "state_snapshot", postgresql.JSONB(astext_type=sa.Text()), nullable=True ), sa.Column( "created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column( "updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True), sa.CheckConstraint( "session_type in ('chat', 'automation')", name="ck_sessions_session_type", ), sa.CheckConstraint( "status in ('pending', 'running', 'completed', 'failed')", name="ck_sessions_status", ), sa.CheckConstraint( "message_count >= 0", name="ck_sessions_message_count_non_negative" ), sa.CheckConstraint( "total_tokens >= 0", name="ck_sessions_total_tokens_non_negative" ), sa.CheckConstraint( "total_cost >= 0", name="ck_sessions_total_cost_non_negative" ), sa.ForeignKeyConstraint(["user_id"], ["auth.users.id"], ondelete="CASCADE"), sa.PrimaryKeyConstraint("id"), ) op.create_index("ix_sessions_user_id", "sessions", ["user_id"], unique=False) op.create_index( "ix_sessions_user_activity", "sessions", ["user_id", "last_activity_at"], unique=False, ) _enable_rls("sessions") op.create_table( "messages", sa.Column("id", sa.UUID(), nullable=False), sa.Column("session_id", sa.UUID(), nullable=False), sa.Column("seq", sa.Integer(), nullable=False), sa.Column("role", sa.String(length=20), nullable=False), sa.Column("content", sa.Text(), nullable=False), sa.Column("model_code", sa.String(length=50), nullable=True), sa.Column("tool_name", sa.String(length=100), nullable=True), sa.Column( "input_tokens", sa.Integer(), server_default=sa.text("0"), nullable=False ), sa.Column( "output_tokens", sa.Integer(), server_default=sa.text("0"), nullable=False ), sa.Column( "cost", sa.Numeric(12, 6), server_default=sa.text("0"), nullable=False ), sa.Column("latency_ms", sa.Integer(), nullable=True), sa.Column( "visibility_mask", sa.BigInteger(), server_default=sa.text("0"), nullable=False, ), sa.Column("metadata", postgresql.JSONB(astext_type=sa.Text()), nullable=True), sa.Column( "created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column( "updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column("deleted_at", sa.DateTime(timezone=True), nullable=True), sa.CheckConstraint("seq > 0", name="ck_messages_seq_positive"), sa.CheckConstraint( "role in ('user', 'assistant', 'system', 'tool')", name="ck_messages_role", ), sa.CheckConstraint( "input_tokens >= 0", name="ck_messages_input_tokens_non_negative" ), sa.CheckConstraint( "output_tokens >= 0", name="ck_messages_output_tokens_non_negative" ), sa.CheckConstraint("cost >= 0", name="ck_messages_cost_non_negative"), sa.CheckConstraint( "latency_ms is null or latency_ms >= 0", name="ck_messages_latency_non_negative", ), sa.ForeignKeyConstraint(["session_id"], ["sessions.id"], ondelete="CASCADE"), sa.PrimaryKeyConstraint("id"), sa.UniqueConstraint("session_id", "seq", name="uq_messages_session_seq"), ) op.create_index("ix_messages_session_id", "messages", ["session_id"], unique=False) op.create_index( "ix_messages_session_seq_visibility", "messages", ["session_id", "seq", "visibility_mask"], unique=False, ) _enable_rls("messages") op.create_table( "user_points", sa.Column("user_id", sa.UUID(), nullable=False), sa.Column( "balance", sa.BigInteger(), server_default=sa.text("0"), nullable=False ), sa.Column( "frozen_balance", sa.BigInteger(), server_default=sa.text("0"), nullable=False, ), sa.Column( "lifetime_earned", sa.BigInteger(), server_default=sa.text("0"), nullable=False, ), sa.Column( "lifetime_spent", sa.BigInteger(), server_default=sa.text("0"), nullable=False, ), sa.Column("version", sa.Integer(), server_default=sa.text("0"), nullable=False), sa.Column( "created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column( "updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.CheckConstraint("balance >= 0", name="ck_user_points_balance_non_negative"), sa.CheckConstraint( "frozen_balance >= 0", name="ck_user_points_frozen_balance_non_negative" ), sa.CheckConstraint( "lifetime_earned >= 0", name="ck_user_points_lifetime_earned_non_negative" ), sa.CheckConstraint( "lifetime_spent >= 0", name="ck_user_points_lifetime_spent_non_negative" ), sa.CheckConstraint( "frozen_balance <= balance", name="ck_user_points_frozen_le_balance" ), sa.ForeignKeyConstraint(["user_id"], ["auth.users.id"], ondelete="CASCADE"), sa.PrimaryKeyConstraint("user_id"), ) _enable_rls("user_points") op.create_table( "points_ledger", sa.Column("id", sa.UUID(), nullable=False), sa.Column("user_id", sa.UUID(), nullable=False), sa.Column("direction", sa.SmallInteger(), nullable=False), sa.Column("amount", sa.BigInteger(), nullable=False), sa.Column("balance_after", sa.BigInteger(), nullable=False), sa.Column("change_type", sa.String(length=16), nullable=False), sa.Column("biz_type", sa.String(length=16), nullable=False), sa.Column("biz_id", sa.UUID(), nullable=False), sa.Column("event_id", sa.String(length=64), nullable=False), sa.Column("operator_id", sa.UUID(), nullable=True), sa.Column( "metadata", postgresql.JSONB(astext_type=sa.Text()), nullable=False, server_default=sa.text("'{}'::jsonb"), ), sa.Column( "created_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.Column( "updated_at", sa.DateTime(timezone=True), server_default=sa.text("now()"), nullable=False, ), sa.CheckConstraint("amount > 0", name="ck_points_ledger_amount_positive"), sa.CheckConstraint( "direction in (1, -1)", name="ck_points_ledger_direction_valid" ), sa.CheckConstraint( "balance_after >= 0", name="ck_points_ledger_balance_after_non_negative" ), sa.CheckConstraint( "change_type in ('register', 'consume', 'grant', 'adjust')", name="ck_points_ledger_change_type", ), sa.CheckConstraint("biz_type = 'chat'", name="ck_points_ledger_biz_type"), sa.CheckConstraint( "((change_type in ('register', 'grant') and direction = 1) " "or (change_type = 'consume' and direction = -1) " "or (change_type = 'adjust' and direction in (1, -1)))", name="ck_points_ledger_direction_by_change_type", ), sa.CheckConstraint( "jsonb_typeof(metadata) = 'object'", name="ck_points_ledger_metadata_object", ), sa.CheckConstraint( "metadata->>'schema_version' = '1' and " "metadata->>'reason_code' in ('REGISTER_WELCOME', 'CHAT_CONSUME', 'CHAT_GRANT', 'CHAT_ADJUST') and " "metadata->>'operator_type' in ('user', 'system', 'admin') and " "coalesce(metadata->>'run_id', '') <> '' and " "(not (metadata ? 'ext') or jsonb_typeof(metadata->'ext') = 'object')", name="ck_points_ledger_metadata_common", ), sa.CheckConstraint( "(change_type <> 'register' or " "(metadata->>'reason_code' = 'REGISTER_WELCOME' and not (metadata ? 'charge')))", name="ck_points_ledger_metadata_register_shape", ), sa.CheckConstraint( "(change_type <> 'consume' or " "(metadata->>'reason_code' = 'CHAT_CONSUME' and " "(metadata ? 'charge') and jsonb_typeof(metadata->'charge') = 'object' and " "(metadata->'charge' ? 'message_id') and (metadata->'charge' ? 'message_seq') and " "(metadata->'charge' ? 'model_code') and (metadata->'charge' ? 'input_tokens') and " "(metadata->'charge' ? 'output_tokens') and (metadata->'charge' ? 'cost')))", name="ck_points_ledger_metadata_consume_shape", ), sa.CheckConstraint( "(change_type <> 'grant' or metadata->>'reason_code' = 'CHAT_GRANT')", name="ck_points_ledger_metadata_grant_shape", ), sa.CheckConstraint( "(change_type <> 'adjust' or " "(metadata->>'reason_code' = 'CHAT_ADJUST' and " "(metadata ? 'ext') and (metadata->'ext' ? 'ticket_id') and " "coalesce(metadata #>> '{ext,ticket_id}', '') <> ''))", name="ck_points_ledger_metadata_adjust_shape", ), sa.ForeignKeyConstraint(["biz_id"], ["sessions.id"], ondelete="RESTRICT"), sa.ForeignKeyConstraint( ["operator_id"], ["auth.users.id"], ondelete="SET NULL" ), sa.ForeignKeyConstraint(["user_id"], ["auth.users.id"], ondelete="CASCADE"), sa.PrimaryKeyConstraint("id"), sa.UniqueConstraint("user_id", "event_id", name="uq_points_ledger_user_event"), ) op.create_index( "ix_points_ledger_user_created_at", "points_ledger", ["user_id", sa.text("created_at DESC")], unique=False, ) op.create_index( "ix_points_ledger_biz_type_biz_id", "points_ledger", ["biz_type", "biz_id"], unique=False, ) _enable_rls("points_ledger") def downgrade() -> None: _drop_rls("points_ledger") op.drop_index("ix_points_ledger_biz_type_biz_id", table_name="points_ledger") op.drop_index("ix_points_ledger_user_created_at", table_name="points_ledger") op.drop_table("points_ledger") _drop_rls("user_points") op.drop_table("user_points") _drop_rls("messages") op.drop_index("ix_messages_session_seq_visibility", table_name="messages") op.drop_index("ix_messages_session_id", table_name="messages") op.drop_table("messages") _drop_rls("sessions") op.drop_index("ix_sessions_user_activity", table_name="sessions") op.drop_index("ix_sessions_user_id", table_name="sessions") op.drop_table("sessions") _drop_rls("profiles") op.drop_index("ix_profiles_settings_gin", table_name="profiles") op.drop_index("ix_profiles_username", table_name="profiles") op.drop_table("profiles") def _enable_rls(table_name: str) -> None: for role in ["anon", "authenticated"]: for action in ["select", "insert", "update", "delete"]: op.execute( f"DROP POLICY IF EXISTS {role}_{action}_{table_name} ON {table_name}" ) op.execute(f"ALTER TABLE {table_name} ENABLE ROW LEVEL SECURITY") for role in ["anon", "authenticated"]: op.execute( f"CREATE POLICY {role}_select_{table_name} ON {table_name} FOR SELECT TO {role} USING (false)" ) op.execute( f"CREATE POLICY {role}_insert_{table_name} ON {table_name} FOR INSERT TO {role} WITH CHECK (false)" ) op.execute( f"CREATE POLICY {role}_update_{table_name} ON {table_name} FOR UPDATE TO {role} USING (false) WITH CHECK (false)" ) op.execute( f"CREATE POLICY {role}_delete_{table_name} ON {table_name} FOR DELETE TO {role} USING (false)" ) def _drop_rls(table_name: str) -> None: for role in ["anon", "authenticated"]: op.execute(f"DROP POLICY IF EXISTS {role}_delete_{table_name} ON {table_name}") op.execute(f"DROP POLICY IF EXISTS {role}_update_{table_name} ON {table_name}") op.execute(f"DROP POLICY IF EXISTS {role}_insert_{table_name} ON {table_name}") op.execute(f"DROP POLICY IF EXISTS {role}_select_{table_name} ON {table_name}") op.execute(f"ALTER TABLE {table_name} DISABLE ROW LEVEL SECURITY")