1

2026-10-25 11:46:44 +08:00
parent 893dfed0ed
commit 494f667719
63 changed files with 5213 additions and 239 deletions
@@ -0,0 +1,49 @@
+"""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')