qzl
55 lines
1.3 KiB
Python
55 lines
1.3 KiB
Python
"""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'))
|
|
"""
|
|
)
|