agent-platform/services/team-service/alembic/versions/20260429_9001_remove_team_versioning.py

"""Remove business version schema artifacts.

Revision ID: 20260429_9001_team
Revises: 20260425_0001
Create Date: 2026-04-29 00:00:00.000000
"""

from alembic import op

revision: str = "20260429_9001_team"
down_revision: str | None = "20260425_0001"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.execute("DO $$\nBEGIN\n    IF to_regclass('team_version') IS NOT NULL AND to_regclass('team_config') IS NULL THEN\n        ALTER TABLE team_version RENAME TO team_config;\n    END IF;\n    IF EXISTS (SELECT 1 FROM information_schema.columns WHERE table_name = 'team_run' AND column_name = 'team_version_id') THEN\n        ALTER TABLE team_run RENAME COLUMN team_version_id TO team_config_id;\n    END IF;\nEND $$;\nALTER TABLE IF EXISTS team_config DROP COLUMN IF EXISTS version_no;\nALTER TABLE IF EXISTS team_config DROP COLUMN IF EXISTS status;\nALTER TABLE IF EXISTS team_config DROP COLUMN IF EXISTS published_time;\nDO $$\nDECLARE\n    table_record record;\nBEGIN\n    FOR table_record IN\n        SELECT table_name\n        FROM information_schema.columns\n        WHERE table_schema = current_schema()\n          AND column_name = 'version'\n    LOOP\n        EXECUTE format('ALTER TABLE %I DROP COLUMN IF EXISTS version', table_record.table_name);\n    END LOOP;\nEND $$;")


def downgrade() -> None:
    # Business version tables and columns were intentionally removed.
    pass