84 lines
2.2 KiB
Python
84 lines
2.2 KiB
Python
"""用户相关 Pydantic v2 Schema（请求/响应）。"""
|
|
|
|
import re
|
|
|
|
from pydantic import BaseModel, EmailStr, Field, field_validator
|
|
|
|
_USERNAME_RE = re.compile(r"^[A-Za-z0-9_\-一-鿿]{2,32}$")
|
|
|
|
|
|
class RegisterRequest(BaseModel):
|
|
username: str = Field(min_length=2, max_length=32)
|
|
email: str = Field(min_length=5, max_length=255)
|
|
password: str = Field(min_length=8, max_length=128)
|
|
|
|
@field_validator("username")
|
|
@classmethod
|
|
def validate_username(cls, v: str) -> str:
|
|
if not _USERNAME_RE.match(v):
|
|
raise ValueError("用户名只能包含字母、数字、下划线、短横线和中文，长度 2-32。")
|
|
return v
|
|
|
|
@field_validator("email")
|
|
@classmethod
|
|
def validate_email(cls, v: str) -> str:
|
|
# 简单格式校验（不引入 email-validator 重依赖）
|
|
if "@" not in v or "." not in v.split("@")[-1]:
|
|
raise ValueError("邮箱格式不正确。")
|
|
return v.lower().strip()
|
|
|
|
@field_validator("password")
|
|
@classmethod
|
|
def validate_password(cls, v: str) -> str:
|
|
if len(v) < 8:
|
|
raise ValueError("密码长度不能少于 8 位。")
|
|
return v
|
|
|
|
|
|
class LoginRequest(BaseModel):
|
|
username_or_email: str = Field(min_length=2, max_length=255)
|
|
password: str = Field(min_length=1, max_length=128)
|
|
|
|
|
|
class UserResponse(BaseModel):
|
|
id: str
|
|
username: str
|
|
email: str
|
|
status: str
|
|
role: str
|
|
storage_used: int
|
|
storage_quota: int
|
|
created_at: str
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class MeResponse(BaseModel):
|
|
id: str
|
|
username: str
|
|
email: str
|
|
status: str
|
|
role: str
|
|
storage_used: int
|
|
storage_quota: int
|
|
created_at: str
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class StorageInfoResponse(BaseModel):
|
|
storage_used: int
|
|
storage_quota: int
|
|
storage_used_mb: float
|
|
storage_quota_mb: float
|
|
|
|
|
|
class UpdateMeRequest(BaseModel):
|
|
password: str | None = Field(default=None, min_length=8, max_length=128)
|
|
|
|
@field_validator("password")
|
|
@classmethod
|
|
def validate_password(cls, v: str | None) -> str | None:
|
|
if v is not None and len(v) < 8:
|
|
raise ValueError("密码长度不能少于 8 位。")
|
|
return v |