50 lines
2.1 KiB
Python
50 lines
2.1 KiB
Python
"""add sources table + source_id on videos/series/collection_sources
|
|
|
|
Revision ID: f1b7d4a9c310
|
|
Revises: e7a1f3c58d02
|
|
Create Date: 2026-10-24 10:00:00.000000
|
|
"""
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import mysql
|
|
|
|
revision = 'f1b7d4a9c310'
|
|
down_revision = 'e7a1f3c58d02'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# 内容来源表（片商）
|
|
op.create_table(
|
|
'sources',
|
|
sa.Column('id', mysql.INTEGER(unsigned=True), autoincrement=True, nullable=False),
|
|
sa.Column('name', sa.String(length=50), nullable=False, comment='来源名称，如 麻豆'),
|
|
sa.Column('code', sa.String(length=50), nullable=False, server_default='', comment='英文/拼音标识'),
|
|
sa.Column('sort', sa.Integer(), nullable=False, server_default='0', comment='越小越靠前'),
|
|
sa.Column('status', mysql.TINYINT(), nullable=False, server_default='1', comment='1启用 0停用'),
|
|
sa.Column('created_at', sa.DateTime(), nullable=False, server_default=sa.func.current_timestamp()),
|
|
sa.PrimaryKeyConstraint('id'),
|
|
sa.UniqueConstraint('name'),
|
|
)
|
|
|
|
# 存量数据全部归为「麻豆」
|
|
op.execute("INSERT INTO sources (name, code, sort, status) VALUES ('麻豆', 'madou', 1, 1)")
|
|
|
|
for table in ('videos', 'series', 'collection_sources'):
|
|
op.add_column(table, sa.Column('source_id', mysql.INTEGER(unsigned=True),
|
|
nullable=False, server_default='0', comment='内容来源 id'))
|
|
op.execute(f"UPDATE {table} SET source_id = (SELECT id FROM sources WHERE code = 'madou')")
|
|
|
|
op.create_index('idx_video_source', 'videos', ['source_id', 'status'])
|
|
op.create_index('idx_series_source', 'series', ['source_id', 'status'])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index('idx_series_source', table_name='series')
|
|
op.drop_index('idx_video_source', table_name='videos')
|
|
op.drop_column('collection_sources', 'source_id')
|
|
op.drop_column('series', 'source_id')
|
|
op.drop_column('videos', 'source_id')
|
|
op.drop_table('sources')
|