import uuid from pydantic import EmailStr from sqlmodel import Field, Relationship, SQLModel # 共享属性 class UserBase(SQLModel): email: EmailStr = Field(unique=True, index=True, max_length=255) is_active: bool = True is_superuser: bool = False full_name: str | None = Field(default=None, max_length=255) # 创建用户时通过 API 接收的属性 class UserCreate(UserBase): password: str = Field(min_length=8, max_length=128) class UserRegister(SQLModel): email: EmailStr = Field(max_length=255) password: str = Field(min_length=8, max_length=128) full_name: str | None = Field(default=None, max_length=255) # 更新用户时通过 API 接收的属性,所有字段均为可选 class UserUpdate(UserBase): email: EmailStr | None = Field(default=None, max_length=255) # type: ignore password: str | None = Field(default=None, min_length=8, max_length=128) class UserUpdateMe(SQLModel): full_name: str | None = Field(default=None, max_length=255) email: EmailStr | None = Field(default=None, max_length=255) class UpdatePassword(SQLModel): current_password: str = Field(min_length=8, max_length=128) new_password: str = Field(min_length=8, max_length=128) # 数据库模型,数据库表名从类名推断 class User(UserBase, table=True): id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True) hashed_password: str items: list["Item"] = Relationship(back_populates="owner", cascade_delete=True) # 通过 API 返回的属性,id 始终必填 class UserPublic(UserBase): id: uuid.UUID class UsersPublic(SQLModel): data: list[UserPublic] count: int # 共享属性 class ItemBase(SQLModel): title: str = Field(min_length=1, max_length=255) description: str | None = Field(default=None, max_length=255) # 创建 Item 时接收的属性 class ItemCreate(ItemBase): pass # 更新 Item 时接收的属性 class ItemUpdate(ItemBase): title: str | None = Field(default=None, min_length=1, max_length=255) # type: ignore # 数据库模型,数据库表名从类名推断 class Item(ItemBase, table=True): id: uuid.UUID = Field(default_factory=uuid.uuid4, primary_key=True) owner_id: uuid.UUID = Field( foreign_key="user.id", nullable=False, ondelete="CASCADE" ) owner: User | None = Relationship(back_populates="items") # 通过 API 返回的属性,id 始终必填 class ItemPublic(ItemBase): id: uuid.UUID owner_id: uuid.UUID class ItemsPublic(SQLModel): data: list[ItemPublic] count: int # 通用消息 class Message(SQLModel): message: str # 包含访问令牌的 JSON 载荷 class Token(SQLModel): access_token: str token_type: str = "bearer" # JWT 令牌内容 class TokenPayload(SQLModel): sub: str | None = None class NewPassword(SQLModel): token: str new_password: str = Field(min_length=8, max_length=128)