mirror of
https://git.victorphan.net/basketballcantho/ten-project.git
synced 2026-08-05 18:33:11 +07:00
97 lines
2.4 KiB
Python
97 lines
2.4 KiB
Python
from datetime import datetime
|
||
from pydantic import BaseModel, EmailStr, field_validator
|
||
import re
|
||
|
||
from .models import UserRole, UserStatus
|
||
|
||
|
||
# ── Auth ─────────────────────────────────────────────────────────────────────
|
||
|
||
class UserRegister(BaseModel):
|
||
username: str
|
||
email: EmailStr
|
||
password: str
|
||
role: UserRole = UserRole.student
|
||
|
||
@field_validator("username")
|
||
@classmethod
|
||
def username_valid(cls, v: str) -> str:
|
||
v = v.strip()
|
||
if not re.match(r"^[A-Za-z0-9_]{3,64}$", v):
|
||
raise ValueError(
|
||
"Username must be 3–64 characters, letters/numbers/underscore only."
|
||
)
|
||
return v
|
||
|
||
@field_validator("password")
|
||
@classmethod
|
||
def password_strength(cls, v: str) -> str:
|
||
if len(v) < 8:
|
||
raise ValueError("Password must be at least 8 characters.")
|
||
return v
|
||
|
||
|
||
class UserLogin(BaseModel):
|
||
username: str
|
||
password: str
|
||
|
||
|
||
class UserOut(BaseModel):
|
||
id: int
|
||
username: str
|
||
email: str
|
||
role: UserRole
|
||
status: UserStatus
|
||
created_at: datetime
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class UserApprove(BaseModel):
|
||
status: UserStatus
|
||
|
||
|
||
class UserRoleUpdate(BaseModel):
|
||
role: UserRole
|
||
|
||
|
||
class TokenPayload(BaseModel):
|
||
sub: int # user id
|
||
exp: int
|
||
|
||
|
||
# ── PDFs ──────────────────────────────────────────────────────────────────────
|
||
|
||
class PDFOut(BaseModel):
|
||
id: int
|
||
user_id: int
|
||
owner_username: str
|
||
title: str
|
||
total_pages: int | None
|
||
created_at: datetime
|
||
|
||
model_config = {"from_attributes": True}
|
||
|
||
|
||
class PDFUploadResult(BaseModel):
|
||
filename: str
|
||
success: bool
|
||
pdf: PDFOut | None = None
|
||
error: str | None = None
|
||
|
||
|
||
# ── Annotations ───────────────────────────────────────────────────────────────
|
||
|
||
class AnnotationIn(BaseModel):
|
||
canvas_data: dict
|
||
|
||
|
||
class AnnotationOut(BaseModel):
|
||
id: int
|
||
pdf_id: int
|
||
page_number: int
|
||
canvas_data: dict
|
||
updated_at: datetime
|
||
|
||
model_config = {"from_attributes": True}
|