feat: 添加后端核心模块
用户认证： - User 模型（支持邮箱/手机号登录） - 双 Token JWT 认证（accessToken + refreshToken） - 注册/登录/刷新 Token API 组织模型： - Brand（品牌方）、Agency（代理商）、Creator（达人） - 多对多关系：品牌方↔代理商、代理商↔达人 项目与任务： - Project 模型（品牌方发布） - Task 模型（完整审核流程追踪） - Brief 模型（解析后的结构化内容） 文件上传： - 阿里云 OSS 直传签名服务 - 支持分片上传，最大 500MB 数据库迁移： - 003_user_org_project_task.py Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
Your Name
co-authored by
Claude Opus 4.5
Claude Opus 4.5
parent
4b8809efe2
commit
4caafdb50f
@@ -3,21 +3,49 @@
|
||||
导出所有 ORM 模型
|
||||
"""
|
||||
from app.models.base import Base, TimestampMixin
|
||||
from app.models.tenant import Tenant
|
||||
from app.models.user import User, UserRole
|
||||
from app.models.organization import Brand, Agency, Creator, brand_agency_association, agency_creator_association
|
||||
from app.models.project import Project, project_agency_association
|
||||
from app.models.task import Task, TaskStage, TaskStatus
|
||||
from app.models.brief import Brief
|
||||
from app.models.ai_config import AIConfig
|
||||
from app.models.review import ReviewTask, ManualTask
|
||||
from app.models.review import ReviewTask, ManualTask, Platform
|
||||
from app.models.rule import ForbiddenWord, WhitelistItem, Competitor
|
||||
from app.models.risk_exception import RiskException
|
||||
|
||||
# 保留 Tenant 兼容旧代码，但新代码应使用 Brand
|
||||
from app.models.tenant import Tenant
|
||||
|
||||
__all__ = [
|
||||
# Base
|
||||
"Base",
|
||||
"TimestampMixin",
|
||||
"Tenant",
|
||||
# 用户与组织
|
||||
"User",
|
||||
"UserRole",
|
||||
"Brand",
|
||||
"Agency",
|
||||
"Creator",
|
||||
"brand_agency_association",
|
||||
"agency_creator_association",
|
||||
# 项目与任务
|
||||
"Project",
|
||||
"project_agency_association",
|
||||
"Task",
|
||||
"TaskStage",
|
||||
"TaskStatus",
|
||||
"Brief",
|
||||
# AI 配置
|
||||
"AIConfig",
|
||||
# 审核（旧模型，保留兼容）
|
||||
"ReviewTask",
|
||||
"ManualTask",
|
||||
"Platform",
|
||||
# 规则
|
||||
"ForbiddenWord",
|
||||
"WhitelistItem",
|
||||
"Competitor",
|
||||
"RiskException",
|
||||
# 兼容
|
||||
"Tenant",
|
||||
]
|
||||
|
||||
@@ -0,0 +1,60 @@
|
||||
"""
|
||||
Brief 模型
|
||||
"""
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
from sqlalchemy import String, Text, ForeignKey
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.models.base import Base, TimestampMixin
|
||||
from app.models.types import JSONType
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.project import Project
|
||||
|
||||
|
||||
class Brief(Base, TimestampMixin):
|
||||
"""Brief 文档表"""
|
||||
__tablename__ = "briefs"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
||||
project_id: Mapped[str] = mapped_column(
|
||||
String(64),
|
||||
ForeignKey("projects.id", ondelete="CASCADE"),
|
||||
unique=True,
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
|
||||
# 原始文件
|
||||
file_url: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
file_name: Mapped[Optional[str]] = mapped_column(String(255), nullable=True)
|
||||
|
||||
# 解析后的结构化内容
|
||||
# 卖点要求: [{"content": "SPF50+", "required": true}, ...]
|
||||
selling_points: Mapped[Optional[list]] = mapped_column(JSONType, nullable=True)
|
||||
|
||||
# 违禁词: [{"word": "最好", "reason": "绝对化用语"}, ...]
|
||||
blacklist_words: Mapped[Optional[list]] = mapped_column(JSONType, nullable=True)
|
||||
|
||||
# 竞品: ["竞品A", "竞品B", ...]
|
||||
competitors: Mapped[Optional[list]] = mapped_column(JSONType, nullable=True)
|
||||
|
||||
# 品牌调性要求
|
||||
brand_tone: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
|
||||
# 时长要求（秒）
|
||||
min_duration: Mapped[Optional[int]] = mapped_column(nullable=True)
|
||||
max_duration: Mapped[Optional[int]] = mapped_column(nullable=True)
|
||||
|
||||
# 其他要求（自由文本）
|
||||
other_requirements: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
|
||||
# 附件文档（代理商上传的参考资料）
|
||||
# [{"id": "af1", "name": "达人拍摄指南.pdf", "url": "...", "size": "1.5MB"}, ...]
|
||||
attachments: Mapped[Optional[list]] = mapped_column(JSONType, nullable=True)
|
||||
|
||||
# 关联
|
||||
project: Mapped["Project"] = relationship("Project", back_populates="brief")
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Brief(id={self.id}, project_id={self.project_id})>"
|
||||
@@ -0,0 +1,160 @@
|
||||
"""
|
||||
组织模型：品牌方、代理商、达人
|
||||
"""
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
from datetime import datetime
|
||||
from sqlalchemy import String, Boolean, Text, ForeignKey, DateTime, Table, Column
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.models.base import Base, TimestampMixin
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.user import User
|
||||
from app.models.project import Project
|
||||
|
||||
|
||||
# 品牌方-代理商 关联表（多对多）
|
||||
brand_agency_association = Table(
|
||||
"brand_agency",
|
||||
Base.metadata,
|
||||
Column("brand_id", String(64), ForeignKey("brands.id", ondelete="CASCADE"), primary_key=True),
|
||||
Column("agency_id", String(64), ForeignKey("agencies.id", ondelete="CASCADE"), primary_key=True),
|
||||
Column("created_at", DateTime(timezone=True), default=datetime.utcnow),
|
||||
Column("is_active", Boolean, default=True),
|
||||
)
|
||||
|
||||
# 代理商-达人 关联表（多对多）
|
||||
agency_creator_association = Table(
|
||||
"agency_creator",
|
||||
Base.metadata,
|
||||
Column("agency_id", String(64), ForeignKey("agencies.id", ondelete="CASCADE"), primary_key=True),
|
||||
Column("creator_id", String(64), ForeignKey("creators.id", ondelete="CASCADE"), primary_key=True),
|
||||
Column("created_at", DateTime(timezone=True), default=datetime.utcnow),
|
||||
Column("is_active", Boolean, default=True),
|
||||
)
|
||||
|
||||
|
||||
class Brand(Base, TimestampMixin):
|
||||
"""品牌方表（即租户）"""
|
||||
__tablename__ = "brands"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(64), primary_key=True) # 格式: BR123456
|
||||
user_id: Mapped[str] = mapped_column(
|
||||
String(64),
|
||||
ForeignKey("users.id", ondelete="CASCADE"),
|
||||
unique=True,
|
||||
nullable=False,
|
||||
)
|
||||
|
||||
# 品牌信息
|
||||
name: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
logo: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
|
||||
# 联系信息
|
||||
contact_name: Mapped[Optional[str]] = mapped_column(String(100), nullable=True)
|
||||
contact_phone: Mapped[Optional[str]] = mapped_column(String(20), nullable=True)
|
||||
contact_email: Mapped[Optional[str]] = mapped_column(String(255), nullable=True)
|
||||
|
||||
# 设置
|
||||
final_review_enabled: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) # 终审开关
|
||||
|
||||
# 状态
|
||||
is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
||||
|
||||
# 关联
|
||||
user: Mapped["User"] = relationship("User", back_populates="brand")
|
||||
agencies: Mapped[list["Agency"]] = relationship(
|
||||
"Agency",
|
||||
secondary=brand_agency_association,
|
||||
back_populates="brands",
|
||||
)
|
||||
projects: Mapped[list["Project"]] = relationship(
|
||||
"Project",
|
||||
back_populates="brand",
|
||||
)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Brand(id={self.id}, name={self.name})>"
|
||||
|
||||
|
||||
class Agency(Base, TimestampMixin):
|
||||
"""代理商表"""
|
||||
__tablename__ = "agencies"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(64), primary_key=True) # 格式: AG123456
|
||||
user_id: Mapped[str] = mapped_column(
|
||||
String(64),
|
||||
ForeignKey("users.id", ondelete="CASCADE"),
|
||||
unique=True,
|
||||
nullable=False,
|
||||
)
|
||||
|
||||
# 代理商信息
|
||||
name: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
logo: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
|
||||
# 联系信息
|
||||
contact_name: Mapped[Optional[str]] = mapped_column(String(100), nullable=True)
|
||||
contact_phone: Mapped[Optional[str]] = mapped_column(String(20), nullable=True)
|
||||
contact_email: Mapped[Optional[str]] = mapped_column(String(255), nullable=True)
|
||||
|
||||
# 权限设置（可被品牌方覆盖）
|
||||
force_pass_enabled: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False) # 强制通过权
|
||||
|
||||
# 状态
|
||||
is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
||||
|
||||
# 关联
|
||||
user: Mapped["User"] = relationship("User", back_populates="agency")
|
||||
brands: Mapped[list["Brand"]] = relationship(
|
||||
"Brand",
|
||||
secondary=brand_agency_association,
|
||||
back_populates="agencies",
|
||||
)
|
||||
creators: Mapped[list["Creator"]] = relationship(
|
||||
"Creator",
|
||||
secondary=agency_creator_association,
|
||||
back_populates="agencies",
|
||||
)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Agency(id={self.id}, name={self.name})>"
|
||||
|
||||
|
||||
class Creator(Base, TimestampMixin):
|
||||
"""达人表"""
|
||||
__tablename__ = "creators"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(64), primary_key=True) # 格式: CR123456
|
||||
user_id: Mapped[str] = mapped_column(
|
||||
String(64),
|
||||
ForeignKey("users.id", ondelete="CASCADE"),
|
||||
unique=True,
|
||||
nullable=False,
|
||||
)
|
||||
|
||||
# 达人信息
|
||||
name: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
avatar: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
bio: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
|
||||
# 社交账号
|
||||
douyin_account: Mapped[Optional[str]] = mapped_column(String(100), nullable=True)
|
||||
xiaohongshu_account: Mapped[Optional[str]] = mapped_column(String(100), nullable=True)
|
||||
bilibili_account: Mapped[Optional[str]] = mapped_column(String(100), nullable=True)
|
||||
|
||||
# 状态
|
||||
is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
||||
|
||||
# 关联
|
||||
user: Mapped["User"] = relationship("User", back_populates="creator")
|
||||
agencies: Mapped[list["Agency"]] = relationship(
|
||||
"Agency",
|
||||
secondary=agency_creator_association,
|
||||
back_populates="creators",
|
||||
)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Creator(id={self.id}, name={self.name})>"
|
||||
@@ -0,0 +1,74 @@
|
||||
"""
|
||||
项目模型
|
||||
"""
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
from datetime import datetime
|
||||
from sqlalchemy import String, Text, ForeignKey, DateTime, Table, Column, Boolean
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
|
||||
from app.models.base import Base, TimestampMixin
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.organization import Brand, Agency
|
||||
from app.models.task import Task
|
||||
from app.models.brief import Brief
|
||||
|
||||
|
||||
# 项目-代理商 关联表（一个项目可以分配给多个代理商）
|
||||
project_agency_association = Table(
|
||||
"project_agency",
|
||||
Base.metadata,
|
||||
Column("project_id", String(64), ForeignKey("projects.id", ondelete="CASCADE"), primary_key=True),
|
||||
Column("agency_id", String(64), ForeignKey("agencies.id", ondelete="CASCADE"), primary_key=True),
|
||||
Column("created_at", DateTime(timezone=True), default=datetime.utcnow),
|
||||
Column("is_active", Boolean, default=True),
|
||||
)
|
||||
|
||||
|
||||
class Project(Base, TimestampMixin):
|
||||
"""项目表"""
|
||||
__tablename__ = "projects"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
||||
brand_id: Mapped[str] = mapped_column(
|
||||
String(64),
|
||||
ForeignKey("brands.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
|
||||
# 项目信息
|
||||
name: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
description: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
|
||||
# 时间
|
||||
start_date: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
deadline: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# 状态
|
||||
status: Mapped[str] = mapped_column(
|
||||
String(20),
|
||||
default="active", # active, completed, archived
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
|
||||
# 关联
|
||||
brand: Mapped["Brand"] = relationship("Brand", back_populates="projects")
|
||||
agencies: Mapped[list["Agency"]] = relationship(
|
||||
"Agency",
|
||||
secondary=project_agency_association,
|
||||
backref="projects",
|
||||
)
|
||||
tasks: Mapped[list["Task"]] = relationship(
|
||||
"Task",
|
||||
back_populates="project",
|
||||
)
|
||||
brief: Mapped[Optional["Brief"]] = relationship(
|
||||
"Brief",
|
||||
back_populates="project",
|
||||
uselist=False,
|
||||
)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Project(id={self.id}, name={self.name})>"
|
||||
@@ -0,0 +1,148 @@
|
||||
"""
|
||||
任务模型
|
||||
"""
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
from datetime import datetime
|
||||
from sqlalchemy import String, Integer, Text, ForeignKey, DateTime, Enum as SQLEnum, Boolean
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
import enum
|
||||
|
||||
from app.models.base import Base, TimestampMixin
|
||||
from app.models.types import JSONType
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.project import Project
|
||||
from app.models.organization import Agency, Creator
|
||||
|
||||
|
||||
class TaskStage(str, enum.Enum):
|
||||
"""任务阶段"""
|
||||
SCRIPT_UPLOAD = "script_upload" # 待上传脚本
|
||||
SCRIPT_AI_REVIEW = "script_ai_review" # 脚本 AI 审核中
|
||||
SCRIPT_AGENCY_REVIEW = "script_agency_review" # 脚本代理商审核中
|
||||
SCRIPT_BRAND_REVIEW = "script_brand_review" # 脚本品牌方终审中
|
||||
VIDEO_UPLOAD = "video_upload" # 待上传视频
|
||||
VIDEO_AI_REVIEW = "video_ai_review" # 视频 AI 审核中
|
||||
VIDEO_AGENCY_REVIEW = "video_agency_review" # 视频代理商审核中
|
||||
VIDEO_BRAND_REVIEW = "video_brand_review" # 视频品牌方终审中
|
||||
COMPLETED = "completed" # 已完成
|
||||
REJECTED = "rejected" # 已驳回
|
||||
|
||||
|
||||
class TaskStatus(str, enum.Enum):
|
||||
"""任务状态"""
|
||||
PENDING = "pending" # 待处理
|
||||
PROCESSING = "processing" # 处理中
|
||||
PASSED = "passed" # 通过
|
||||
REJECTED = "rejected" # 驳回
|
||||
FORCE_PASSED = "force_passed" # 强制通过
|
||||
|
||||
|
||||
class Task(Base, TimestampMixin):
|
||||
"""任务表"""
|
||||
__tablename__ = "tasks"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
||||
|
||||
# 关联
|
||||
project_id: Mapped[str] = mapped_column(
|
||||
String(64),
|
||||
ForeignKey("projects.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
agency_id: Mapped[str] = mapped_column(
|
||||
String(64),
|
||||
ForeignKey("agencies.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
creator_id: Mapped[str] = mapped_column(
|
||||
String(64),
|
||||
ForeignKey("creators.id", ondelete="CASCADE"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
|
||||
# 任务信息
|
||||
name: Mapped[str] = mapped_column(String(255), nullable=False) # 如 "宣传任务(1)"
|
||||
sequence: Mapped[int] = mapped_column(Integer, default=1, nullable=False) # 序号
|
||||
|
||||
# 当前阶段
|
||||
stage: Mapped[TaskStage] = mapped_column(
|
||||
SQLEnum(TaskStage, name="task_stage_enum"),
|
||||
default=TaskStage.SCRIPT_UPLOAD,
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
|
||||
# ===== 脚本相关 =====
|
||||
script_file_url: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
script_file_name: Mapped[Optional[str]] = mapped_column(String(255), nullable=True)
|
||||
script_uploaded_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# 脚本 AI 审核结果
|
||||
script_ai_score: Mapped[Optional[int]] = mapped_column(Integer, nullable=True)
|
||||
script_ai_result: Mapped[Optional[dict]] = mapped_column(JSONType, nullable=True)
|
||||
script_ai_reviewed_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# 脚本代理商审核
|
||||
script_agency_status: Mapped[Optional[TaskStatus]] = mapped_column(
|
||||
SQLEnum(TaskStatus, name="task_status_enum"),
|
||||
nullable=True,
|
||||
)
|
||||
script_agency_comment: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
script_agency_reviewer_id: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
|
||||
script_agency_reviewed_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# 脚本品牌方终审
|
||||
script_brand_status: Mapped[Optional[TaskStatus]] = mapped_column(
|
||||
SQLEnum(TaskStatus, name="task_status_enum", create_type=False),
|
||||
nullable=True,
|
||||
)
|
||||
script_brand_comment: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
script_brand_reviewer_id: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
|
||||
script_brand_reviewed_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# ===== 视频相关 =====
|
||||
video_file_url: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
video_file_name: Mapped[Optional[str]] = mapped_column(String(255), nullable=True)
|
||||
video_duration: Mapped[Optional[int]] = mapped_column(Integer, nullable=True) # 秒
|
||||
video_thumbnail_url: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
video_uploaded_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# 视频 AI 审核结果
|
||||
video_ai_score: Mapped[Optional[int]] = mapped_column(Integer, nullable=True)
|
||||
video_ai_result: Mapped[Optional[dict]] = mapped_column(JSONType, nullable=True)
|
||||
video_ai_reviewed_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# 视频代理商审核
|
||||
video_agency_status: Mapped[Optional[TaskStatus]] = mapped_column(
|
||||
SQLEnum(TaskStatus, name="task_status_enum", create_type=False),
|
||||
nullable=True,
|
||||
)
|
||||
video_agency_comment: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
video_agency_reviewer_id: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
|
||||
video_agency_reviewed_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# 视频品牌方终审
|
||||
video_brand_status: Mapped[Optional[TaskStatus]] = mapped_column(
|
||||
SQLEnum(TaskStatus, name="task_status_enum", create_type=False),
|
||||
nullable=True,
|
||||
)
|
||||
video_brand_comment: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
|
||||
video_brand_reviewer_id: Mapped[Optional[str]] = mapped_column(String(64), nullable=True)
|
||||
video_brand_reviewed_at: Mapped[Optional[datetime]] = mapped_column(DateTime(timezone=True), nullable=True)
|
||||
|
||||
# ===== 申诉相关 =====
|
||||
appeal_count: Mapped[int] = mapped_column(Integer, default=1, nullable=False) # 剩余申诉次数
|
||||
is_appeal: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False) # 是否为申诉
|
||||
appeal_reason: Mapped[Optional[str]] = mapped_column(Text, nullable=True) # 申诉理由
|
||||
|
||||
# 关联
|
||||
project: Mapped["Project"] = relationship("Project", back_populates="tasks")
|
||||
agency: Mapped["Agency"] = relationship("Agency", foreign_keys=[agency_id])
|
||||
creator: Mapped["Creator"] = relationship("Creator", foreign_keys=[creator_id])
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<Task(id={self.id}, name={self.name}, stage={self.stage})>"
|
||||
@@ -0,0 +1,80 @@
|
||||
"""
|
||||
用户模型
|
||||
"""
|
||||
from typing import TYPE_CHECKING, Optional
|
||||
from datetime import datetime
|
||||
from sqlalchemy import String, Boolean, DateTime, Enum as SQLEnum
|
||||
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
||||
import enum
|
||||
|
||||
from app.models.base import Base, TimestampMixin
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from app.models.organization import Brand, Agency, Creator
|
||||
|
||||
|
||||
class UserRole(str, enum.Enum):
|
||||
"""用户角色"""
|
||||
BRAND = "brand" # 品牌方
|
||||
AGENCY = "agency" # 代理商
|
||||
CREATOR = "creator" # 达人
|
||||
|
||||
|
||||
class User(Base, TimestampMixin):
|
||||
"""用户表"""
|
||||
__tablename__ = "users"
|
||||
|
||||
id: Mapped[str] = mapped_column(String(64), primary_key=True)
|
||||
|
||||
# 登录凭证（邮箱和手机号都可以登录）
|
||||
email: Mapped[Optional[str]] = mapped_column(String(255), unique=True, nullable=True, index=True)
|
||||
phone: Mapped[Optional[str]] = mapped_column(String(20), unique=True, nullable=True, index=True)
|
||||
password_hash: Mapped[str] = mapped_column(String(255), nullable=False)
|
||||
|
||||
# 用户信息
|
||||
name: Mapped[str] = mapped_column(String(100), nullable=False)
|
||||
avatar: Mapped[Optional[str]] = mapped_column(String(2048), nullable=True)
|
||||
|
||||
# 角色
|
||||
role: Mapped[UserRole] = mapped_column(
|
||||
SQLEnum(UserRole, name="user_role_enum"),
|
||||
nullable=False,
|
||||
index=True,
|
||||
)
|
||||
|
||||
# 状态
|
||||
is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
||||
is_verified: Mapped[bool] = mapped_column(Boolean, default=False, nullable=False)
|
||||
|
||||
# 最后登录
|
||||
last_login_at: Mapped[Optional[datetime]] = mapped_column(
|
||||
DateTime(timezone=True),
|
||||
nullable=True,
|
||||
)
|
||||
|
||||
# Refresh Token（用于 JWT 刷新）
|
||||
refresh_token: Mapped[Optional[str]] = mapped_column(String(512), nullable=True)
|
||||
refresh_token_expires_at: Mapped[Optional[datetime]] = mapped_column(
|
||||
DateTime(timezone=True),
|
||||
nullable=True,
|
||||
)
|
||||
|
||||
# 关联的组织（根据角色不同，关联到不同的组织）
|
||||
brand: Mapped[Optional["Brand"]] = relationship(
|
||||
"Brand",
|
||||
back_populates="user",
|
||||
uselist=False,
|
||||
)
|
||||
agency: Mapped[Optional["Agency"]] = relationship(
|
||||
"Agency",
|
||||
back_populates="user",
|
||||
uselist=False,
|
||||
)
|
||||
creator: Mapped[Optional["Creator"]] = relationship(
|
||||
"Creator",
|
||||
back_populates="user",
|
||||
uselist=False,
|
||||
)
|
||||
|
||||
def __repr__(self) -> str:
|
||||
return f"<User(id={self.id}, email={self.email}, role={self.role})>"
|
||||
Reference in New Issue
Block a user