35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""add data source db_database and db_schema columns
|
|
|
|
Revision ID: g7h8i9j0k1l2
|
|
Revises: f6g7h8i9j0k1
|
|
Create Date: 2026-06-04 14:00:00.000000
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision: str = 'g7h8i9j0k1l2'
|
|
down_revision: Union[str, None] = 'f6g7h8i9j0k1'
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.add_column(
|
|
'core_data_source',
|
|
sa.Column('db_database', sa.String(length=100), nullable=True, server_default='', comment='SQL执行目标库/catalog'),
|
|
)
|
|
op.add_column(
|
|
'core_data_source',
|
|
sa.Column('db_schema', sa.String(length=100), nullable=True, server_default='', comment='SQL执行目标模式/schema'),
|
|
)
|
|
op.execute("UPDATE core_data_source SET db_database = '' WHERE db_database IS NULL")
|
|
op.execute("UPDATE core_data_source SET db_schema = '' WHERE db_schema IS NULL")
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_column('core_data_source', 'db_schema')
|
|
op.drop_column('core_data_source', 'db_database')
|