File size: 3,068 Bytes
35e7795 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 | """用户相关的Pydantic模型"""
from datetime import datetime
from typing import Optional
from pydantic import BaseModel, Field
class UserBase(BaseModel):
"""用户基础模型"""
username: str = Field(..., description="用户名", min_length=3, max_length=50)
full_name: Optional[str] = Field(None, description="全名", max_length=100)
organization: Optional[str] = Field(None, description="单位", max_length=100)
team: Optional[str] = Field(None, description="团队", max_length=100)
species: Optional[str] = Field(None, description="物种", max_length=100)
is_active: bool = Field(True, description="是否激活")
is_superuser: bool = Field(False, description="是否超级用户")
class UserCreate(UserBase):
"""创建用户模型"""
password: str = Field(..., description="密码哈希值（SHA-256）", min_length=64)
class UserUpdate(BaseModel):
"""更新用户模型"""
username: Optional[str] = Field(
None, description="用户名", min_length=3, max_length=50
)
full_name: Optional[str] = Field(None, description="全名", max_length=100)
organization: Optional[str] = Field(None, description="单位", max_length=100)
team: Optional[str] = Field(None, description="团队", max_length=100)
species: Optional[str] = Field(None, description="物种", max_length=100)
password: Optional[str] = Field(
None, description="密码哈希值（SHA-256）", min_length=64
)
is_active: Optional[bool] = Field(None, description="是否激活")
is_superuser: Optional[bool] = Field(None, description="是否超级用户")
class User(UserBase):
"""用户模型"""
id: int = Field(..., description="用户ID")
created_at: datetime = Field(..., description="创建时间")
updated_at: datetime = Field(..., description="更新时间")
class Config:
from_attributes = True
class UserRegister(BaseModel):
"""用户注册模型"""
username: str = Field(..., description="用户名", min_length=3, max_length=50)
password: str = Field(..., description="密码哈希值（SHA-256）", min_length=64)
full_name: Optional[str] = Field(None, description="全名", max_length=100)
organization: Optional[str] = Field(None, description="单位", max_length=100)
team: Optional[str] = Field(None, description="团队", max_length=100)
species: Optional[str] = Field(None, description="物种", max_length=100)
class UserLogin(BaseModel):
"""用户登录模型"""
username: str = Field(..., description="用户名", min_length=3, max_length=50)
password: str = Field(
..., description="密码哈希值（SHA-256+时间戳）", min_length=64
)
timestamp: int = Field(..., description="时间戳（Unix时间戳，秒）")
class Token(BaseModel):
"""登录令牌模型"""
access_token: str = Field(..., description="访问令牌")
token_type: str = Field(default="bearer", description="令牌类型")
user: User = Field(..., description="用户信息")
|