4c9b2f1263
- Brief 支持代理商附件上传 (迁移 007) - 项目新增 platform 字段 (迁移 008)，前端创建/展示平台信息 - 修复 AI 规则解析：处理中文引号导致 JSON 解析失败的问题 - 修复消息中心崩溃：补全后端消息类型映射 + fallback 保护 - 项目创建时自动发送消息通知 - .gitignore 排除 backend/data/ 数据库文件 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
"""
|
||
用户模型
|
||
"""
|
||
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", values_callable=lambda x: [e.value for e in x]),
|
||
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})>"
|