"""Fix tier CHECK constraint to allow all plan tiers

Revision ID: 006
Revises: cb71a958ad92
"""
from alembic import op

revision = "006"
down_revision = "a1b2c3d4e5f6"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.execute(
        "ALTER TABLE users DROP CONSTRAINT IF EXISTS ck_users_tier"
    )
    op.execute(
        "ALTER TABLE users ADD CONSTRAINT ck_users_tier "
        "CHECK (tier IN ('free', 'starter', 'pro', 'business', 'enterprise'))"
    )


def downgrade() -> None:
    op.execute(
        "ALTER TABLE users DROP CONSTRAINT IF EXISTS ck_users_tier"
    )
    op.execute(
        "ALTER TABLE users ADD CONSTRAINT ck_users_tier "
        "CHECK (tier IN ('free', 'pro'))"
    )
