john
89 lines
2.2 KiB
Python
89 lines
2.2 KiB
Python
"""Sync DB with Models
|
|
|
|
Revision ID: 002
|
|
Revises: 001
|
|
Create Date: 2025-10-05 11:28:41.772294
|
|
|
|
"""
|
|
|
|
from enum import Enum
|
|
from typing import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '002'
|
|
down_revision: str | None = '001'
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
class EventCallbackStatus(Enum):
|
|
ACTIVE = 'ACTIVE'
|
|
DISABLED = 'DISABLED'
|
|
COMPLETED = 'COMPLETED'
|
|
ERROR = 'ERROR'
|
|
|
|
|
|
def upgrade() -> None:
|
|
"""Upgrade schema."""
|
|
bind = op.get_bind()
|
|
event_callback_status = sa.Enum(
|
|
EventCallbackStatus,
|
|
name='eventcallbackstatus',
|
|
)
|
|
event_callback_status.create(bind, checkfirst=True)
|
|
|
|
op.add_column(
|
|
'event_callback',
|
|
sa.Column(
|
|
'status',
|
|
event_callback_status,
|
|
nullable=False,
|
|
server_default='ACTIVE',
|
|
),
|
|
)
|
|
op.add_column(
|
|
'event_callback',
|
|
sa.Column(
|
|
'updated_at', sa.DateTime, nullable=False, server_default=sa.func.now()
|
|
),
|
|
)
|
|
op.drop_index('ix_event_callback_result_event_id')
|
|
op.drop_column('event_callback_result', 'event_id')
|
|
op.add_column(
|
|
'event_callback_result', sa.Column('event_id', sa.String, nullable=True)
|
|
)
|
|
op.create_index(
|
|
op.f('ix_event_callback_result_event_id'),
|
|
'event_callback_result',
|
|
['event_id'],
|
|
unique=False,
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
"""Downgrade schema."""
|
|
bind = op.get_bind()
|
|
event_callback_status = sa.Enum(
|
|
EventCallbackStatus,
|
|
name='eventcallbackstatus',
|
|
)
|
|
|
|
op.drop_column('event_callback', 'status')
|
|
op.drop_column('event_callback', 'updated_at')
|
|
op.drop_index('ix_event_callback_result_event_id')
|
|
op.drop_column('event_callback_result', 'event_id')
|
|
op.add_column(
|
|
'event_callback_result', sa.Column('event_id', sa.UUID, nullable=True)
|
|
)
|
|
op.create_index(
|
|
op.f('ix_event_callback_result_event_id'),
|
|
'event_callback_result',
|
|
['event_id'],
|
|
unique=False,
|
|
)
|
|
|
|
event_callback_status.drop(bind, checkfirst=True)
|