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>
90 lines
2.8 KiB
Python
90 lines
2.8 KiB
Python
"""
|
|
审核任务模型
|
|
"""
|
|
from typing import TYPE_CHECKING, Optional
|
|
from datetime import datetime
|
|
from sqlalchemy import String, Integer, Float, Text, ForeignKey, DateTime, Enum as SQLEnum
|
|
from app.models.types import JSONType
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
import enum
|
|
|
|
from app.models.base import Base, TimestampMixin
|
|
|
|
if TYPE_CHECKING:
|
|
from app.models.tenant import Tenant
|
|
|
|
|
|
class TaskStatus(str, enum.Enum):
|
|
"""任务状态"""
|
|
PENDING = "pending"
|
|
PROCESSING = "processing"
|
|
COMPLETED = "completed"
|
|
FAILED = "failed"
|
|
APPROVED = "approved"
|
|
REJECTED = "rejected"
|
|
|
|
|
|
class Platform(str, enum.Enum):
|
|
"""投放平台"""
|
|
DOUYIN = "douyin"
|
|
XIAOHONGSHU = "xiaohongshu"
|
|
BILIBILI = "bilibili"
|
|
KUAISHOU = "kuaishou"
|
|
|
|
|
|
class ReviewTask(Base, TimestampMixin):
|
|
"""审核任务表 (AI 自动审核)"""
|
|
__tablename__ = "review_tasks"
|
|
|
|
id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
|
tenant_id: Mapped[str] = mapped_column(
|
|
String(64),
|
|
ForeignKey("tenants.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
|
|
# 视频信息
|
|
video_url: Mapped[str] = mapped_column(String(2048), nullable=False)
|
|
platform: Mapped[Platform] = mapped_column(
|
|
SQLEnum(Platform, name="platform_enum"),
|
|
nullable=False,
|
|
)
|
|
brand_id: Mapped[str] = mapped_column(String(64), nullable=False, index=True)
|
|
creator_id: Mapped[str] = mapped_column(String(64), nullable=False, index=True)
|
|
|
|
# 审核状态
|
|
status: Mapped[TaskStatus] = mapped_column(
|
|
SQLEnum(TaskStatus, name="task_status_enum"),
|
|
default=TaskStatus.PENDING,
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
progress: Mapped[int] = mapped_column(Integer, default=0, nullable=False)
|
|
current_step: Mapped[str] = mapped_column(String(100), default="等待处理", nullable=False)
|
|
|
|
# 审核结果
|
|
score: Mapped[Optional[int]] = mapped_column(Integer, nullable=True)
|
|
summary: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
|
|
|
# 违规详情 (JSON 数组)
|
|
# [{"type": "forbidden_word", "content": "最好", "severity": "high", ...}]
|
|
violations: Mapped[Optional[list]] = mapped_column(JSONType, nullable=True)
|
|
|
|
# 软性风控提示 (JSON 数组)
|
|
soft_warnings: Mapped[Optional[list]] = mapped_column(JSONType, nullable=True)
|
|
|
|
# 审核要求 (JSON)
|
|
requirements: Mapped[Optional[dict]] = mapped_column(JSONType, nullable=True)
|
|
|
|
# 竞品列表
|
|
competitors: Mapped[Optional[list]] = mapped_column(JSONType, nullable=True)
|
|
|
|
# 错误信息
|
|
error_message: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
|
|
|
# 关联
|
|
tenant: Mapped["Tenant"] = relationship("Tenant", back_populates="review_tasks")
|
|
def __repr__(self) -> str:
|
|
return f"<ReviewTask(id={self.id}, status={self.status})>"
|