Files
social-app/backend/alembic/versions/20260325_0001_schedule_item_status_converge.py
T

55 lines
1.3 KiB
Python
Raw Normal View History

"""converge schedule item status to active or archived
Revision ID: 202603250001
Revises: 202603240001
Create Date: 2026-03-25 10:00:00
"""
from typing import Sequence, Union
from alembic import op
revision: str = "202603250001"
down_revision: Union[str, Sequence[str], None] = "202603240001"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
op.execute(
"""
UPDATE public.schedule_items
SET status = 'archived'
WHERE status IN ('completed', 'canceled')
"""
)
op.execute(
"""
ALTER TABLE public.schedule_items
DROP CONSTRAINT IF EXISTS chk_schedule_item_status
"""
)
op.execute(
"""
ALTER TABLE public.schedule_items
ADD CONSTRAINT chk_schedule_item_status
CHECK (status IN ('active', 'archived'))
"""
)
def downgrade() -> None:
op.execute(
"""
ALTER TABLE public.schedule_items
DROP CONSTRAINT IF EXISTS chk_schedule_item_status
"""
)
op.execute(
"""
ALTER TABLE public.schedule_items
ADD CONSTRAINT chk_schedule_item_status
CHECK (status IN ('active', 'completed', 'canceled', 'archived'))
"""
)