4a3c7e7923
- 删除后端 risk_exceptions 模块（API/Model/Schema/迁移/测试） - 删除后端 metrics 模块（API/测试） - 删除后端 ManualTask 模型和相关 Schema - 修复搜索接口响应缺少 total 字段的问题 - 统一 Platform 枚举（前端去掉后端不支持的 weibo/wechat） - 新增前端注册页面 /register，登录页添加注册链接 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
151 lines
7.3 KiB
Python
151 lines
7.3 KiB
Python
"""初始表结构
|
|
|
|
Revision ID: 001
|
|
Revises:
|
|
Create Date: 2024-01-15
|
|
|
|
"""
|
|
from typing import Sequence, Union
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision: str = '001'
|
|
down_revision: Union[str, None] = None
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# 创建枚举类型
|
|
platform_enum = postgresql.ENUM(
|
|
'douyin', 'xiaohongshu', 'bilibili', 'kuaishou',
|
|
name='platform_enum'
|
|
)
|
|
platform_enum.create(op.get_bind(), checkfirst=True)
|
|
|
|
task_status_enum = postgresql.ENUM(
|
|
'pending', 'processing', 'completed', 'failed', 'approved', 'rejected',
|
|
name='task_status_enum'
|
|
)
|
|
task_status_enum.create(op.get_bind(), checkfirst=True)
|
|
|
|
# 租户表
|
|
op.create_table(
|
|
'tenants',
|
|
sa.Column('id', sa.String(64), primary_key=True),
|
|
sa.Column('name', sa.String(255), nullable=False),
|
|
sa.Column('is_active', sa.Boolean(), nullable=False, default=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now(), onupdate=sa.func.now(), nullable=False),
|
|
)
|
|
|
|
# AI 配置表
|
|
op.create_table(
|
|
'ai_configs',
|
|
sa.Column('id', sa.Integer(), primary_key=True, autoincrement=True),
|
|
sa.Column('tenant_id', sa.String(64), sa.ForeignKey('tenants.id', ondelete='CASCADE'), unique=True, nullable=False),
|
|
sa.Column('provider', sa.String(50), nullable=False),
|
|
sa.Column('base_url', sa.String(500), nullable=False),
|
|
sa.Column('api_key_encrypted', sa.Text(), nullable=False),
|
|
sa.Column('models', postgresql.JSONB(), nullable=False),
|
|
sa.Column('temperature', sa.Float(), nullable=False, default=0.7),
|
|
sa.Column('max_tokens', sa.Integer(), nullable=False, default=2000),
|
|
sa.Column('available_models', postgresql.JSONB(), nullable=True),
|
|
sa.Column('last_test_at', sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column('last_test_result', postgresql.JSONB(), nullable=True),
|
|
sa.Column('is_configured', sa.Boolean(), nullable=False, default=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now(), onupdate=sa.func.now(), nullable=False),
|
|
)
|
|
op.create_index('ix_ai_configs_tenant_id', 'ai_configs', ['tenant_id'])
|
|
|
|
# 审核任务表
|
|
op.create_table(
|
|
'review_tasks',
|
|
sa.Column('id', sa.String(64), primary_key=True),
|
|
sa.Column('tenant_id', sa.String(64), sa.ForeignKey('tenants.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('video_url', sa.String(2048), nullable=False),
|
|
sa.Column('platform', platform_enum, nullable=False),
|
|
sa.Column('brand_id', sa.String(64), nullable=False),
|
|
sa.Column('creator_id', sa.String(64), nullable=False),
|
|
sa.Column('status', task_status_enum, nullable=False, default='pending'),
|
|
sa.Column('progress', sa.Integer(), nullable=False, default=0),
|
|
sa.Column('current_step', sa.String(100), nullable=False, default='等待处理'),
|
|
sa.Column('score', sa.Integer(), nullable=True),
|
|
sa.Column('summary', sa.Text(), nullable=True),
|
|
sa.Column('violations', postgresql.JSONB(), nullable=True),
|
|
sa.Column('soft_warnings', postgresql.JSONB(), nullable=True),
|
|
sa.Column('requirements', postgresql.JSONB(), nullable=True),
|
|
sa.Column('competitors', postgresql.JSONB(), nullable=True),
|
|
sa.Column('error_message', sa.Text(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now(), onupdate=sa.func.now(), nullable=False),
|
|
)
|
|
op.create_index('ix_review_tasks_tenant_id', 'review_tasks', ['tenant_id'])
|
|
op.create_index('ix_review_tasks_brand_id', 'review_tasks', ['brand_id'])
|
|
op.create_index('ix_review_tasks_creator_id', 'review_tasks', ['creator_id'])
|
|
op.create_index('ix_review_tasks_status', 'review_tasks', ['status'])
|
|
|
|
# 违禁词表
|
|
op.create_table(
|
|
'forbidden_words',
|
|
sa.Column('id', sa.String(64), primary_key=True),
|
|
sa.Column('tenant_id', sa.String(64), sa.ForeignKey('tenants.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('word', sa.String(255), nullable=False),
|
|
sa.Column('category', sa.String(100), nullable=False),
|
|
sa.Column('severity', sa.String(50), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now(), onupdate=sa.func.now(), nullable=False),
|
|
)
|
|
op.create_index('ix_forbidden_words_tenant_id', 'forbidden_words', ['tenant_id'])
|
|
op.create_index('ix_forbidden_words_word', 'forbidden_words', ['word'])
|
|
op.create_index('ix_forbidden_words_category', 'forbidden_words', ['category'])
|
|
|
|
# 白名单表
|
|
op.create_table(
|
|
'whitelist_items',
|
|
sa.Column('id', sa.String(64), primary_key=True),
|
|
sa.Column('tenant_id', sa.String(64), sa.ForeignKey('tenants.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('brand_id', sa.String(64), nullable=False),
|
|
sa.Column('term', sa.String(255), nullable=False),
|
|
sa.Column('reason', sa.Text(), nullable=False),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now(), onupdate=sa.func.now(), nullable=False),
|
|
)
|
|
op.create_index('ix_whitelist_items_tenant_id', 'whitelist_items', ['tenant_id'])
|
|
op.create_index('ix_whitelist_items_brand_id', 'whitelist_items', ['brand_id'])
|
|
op.create_index('ix_whitelist_items_term', 'whitelist_items', ['term'])
|
|
|
|
# 竞品表
|
|
op.create_table(
|
|
'competitors',
|
|
sa.Column('id', sa.String(64), primary_key=True),
|
|
sa.Column('tenant_id', sa.String(64), sa.ForeignKey('tenants.id', ondelete='CASCADE'), nullable=False),
|
|
sa.Column('brand_id', sa.String(64), nullable=False),
|
|
sa.Column('name', sa.String(255), nullable=False),
|
|
sa.Column('logo_url', sa.String(2048), nullable=True),
|
|
sa.Column('keywords', postgresql.JSONB(), nullable=True),
|
|
sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.func.now(), nullable=False),
|
|
sa.Column('updated_at', sa.DateTime(timezone=True), server_default=sa.func.now(), onupdate=sa.func.now(), nullable=False),
|
|
)
|
|
op.create_index('ix_competitors_tenant_id', 'competitors', ['tenant_id'])
|
|
op.create_index('ix_competitors_brand_id', 'competitors', ['brand_id'])
|
|
|
|
|
|
|
|
def downgrade() -> None:
|
|
# 删除表
|
|
op.drop_table('competitors')
|
|
op.drop_table('whitelist_items')
|
|
op.drop_table('forbidden_words')
|
|
op.drop_table('review_tasks')
|
|
op.drop_table('ai_configs')
|
|
op.drop_table('tenants')
|
|
|
|
# 删除枚举类型
|
|
op.execute('DROP TYPE IF EXISTS task_status_enum')
|
|
op.execute('DROP TYPE IF EXISTS platform_enum')
|