jlacoste
163 lines
4.7 KiB
Python
163 lines
4.7 KiB
Python
from datetime import datetime
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, field_validator
|
|
|
|
from models import BeerStyle, DrinkCategory, SpiritType, WineColor
|
|
|
|
|
|
def validate_password_strength(v: str) -> str:
|
|
if not any(c.isupper() for c in v):
|
|
raise ValueError("Le mot de passe doit contenir au moins une majuscule")
|
|
if not any(c.islower() for c in v):
|
|
raise ValueError("Le mot de passe doit contenir au moins une minuscule")
|
|
if not any(c.isdigit() for c in v):
|
|
raise ValueError("Le mot de passe doit contenir au moins un chiffre")
|
|
return v
|
|
|
|
|
|
# ── Auth ──
|
|
|
|
class UserCreate(BaseModel):
|
|
username: str = Field(..., min_length=3, max_length=50)
|
|
email: str = Field(..., max_length=255)
|
|
password: str = Field(..., min_length=8, max_length=128)
|
|
invite_token: str = Field(..., min_length=1, max_length=128)
|
|
|
|
@field_validator("password")
|
|
@classmethod
|
|
def validate_password(cls, v: str) -> str:
|
|
return validate_password_strength(v)
|
|
|
|
@field_validator("email")
|
|
@classmethod
|
|
def validate_email(cls, v: str) -> str:
|
|
if "@" not in v or "." not in v.split("@")[-1]:
|
|
raise ValueError("Adresse email invalide")
|
|
return v.lower().strip()
|
|
|
|
|
|
class UserLogin(BaseModel):
|
|
username: str = Field(..., min_length=1, max_length=50)
|
|
password: str = Field(..., min_length=1, max_length=128)
|
|
|
|
|
|
class UserResponse(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
username: str
|
|
email: str
|
|
is_admin: bool
|
|
created_at: datetime
|
|
|
|
|
|
class TokenResponse(BaseModel):
|
|
access_token: str
|
|
token_type: str = "bearer"
|
|
user: UserResponse
|
|
|
|
|
|
class PasswordChange(BaseModel):
|
|
current_password: str = Field(..., min_length=1, max_length=128)
|
|
new_password: str = Field(..., min_length=8, max_length=128)
|
|
|
|
@field_validator("new_password")
|
|
@classmethod
|
|
def validate_password(cls, v: str) -> str:
|
|
return validate_password_strength(v)
|
|
|
|
|
|
class InvitationCreate(BaseModel):
|
|
pass
|
|
|
|
|
|
# ── Admin ──
|
|
|
|
|
|
class AdminResetPassword(BaseModel):
|
|
new_password: str = Field(..., min_length=8, max_length=128)
|
|
|
|
@field_validator("new_password")
|
|
@classmethod
|
|
def validate_password(cls, v: str) -> str:
|
|
return validate_password_strength(v)
|
|
|
|
|
|
class AdminToggleAdmin(BaseModel):
|
|
is_admin: bool
|
|
|
|
|
|
class InvitationResponse(BaseModel):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
token: str
|
|
created_by: int
|
|
used_by: Optional[int] = None
|
|
created_at: datetime
|
|
used_at: Optional[datetime] = None
|
|
expires_at: datetime
|
|
|
|
|
|
# ── Drinks ──
|
|
|
|
class DrinkBase(BaseModel):
|
|
name: str = Field(..., min_length=1, max_length=255)
|
|
category: DrinkCategory
|
|
rating: Optional[float] = Field(None, ge=0, le=5)
|
|
notes: Optional[str] = Field(None, max_length=5000)
|
|
|
|
grape_variety: Optional[str] = Field(None, max_length=255)
|
|
vintage: Optional[int] = Field(None, ge=1900, le=2100)
|
|
region: Optional[str] = Field(None, max_length=255)
|
|
producer: Optional[str] = Field(None, max_length=255)
|
|
wine_color: Optional[WineColor] = None
|
|
|
|
brewery: Optional[str] = Field(None, max_length=255)
|
|
beer_style: Optional[BeerStyle] = None
|
|
ibu: Optional[float] = Field(None, ge=0, le=200)
|
|
abv: Optional[float] = Field(None, ge=0, le=30)
|
|
|
|
spirit_type: Optional[SpiritType] = None
|
|
age_years: Optional[int] = Field(None, ge=0, le=200)
|
|
distillery: Optional[str] = Field(None, max_length=255)
|
|
country: Optional[str] = Field(None, max_length=255)
|
|
|
|
|
|
class DrinkCreate(DrinkBase):
|
|
pass
|
|
|
|
|
|
class DrinkUpdate(BaseModel):
|
|
name: Optional[str] = Field(None, min_length=1, max_length=255)
|
|
category: Optional[DrinkCategory] = None
|
|
rating: Optional[float] = Field(None, ge=0, le=5)
|
|
notes: Optional[str] = Field(None, max_length=5000)
|
|
|
|
grape_variety: Optional[str] = Field(None, max_length=255)
|
|
vintage: Optional[int] = Field(None, ge=1900, le=2100)
|
|
region: Optional[str] = Field(None, max_length=255)
|
|
producer: Optional[str] = Field(None, max_length=255)
|
|
wine_color: Optional[WineColor] = None
|
|
|
|
brewery: Optional[str] = Field(None, max_length=255)
|
|
beer_style: Optional[BeerStyle] = None
|
|
ibu: Optional[float] = Field(None, ge=0, le=200)
|
|
abv: Optional[float] = Field(None, ge=0, le=30)
|
|
|
|
spirit_type: Optional[SpiritType] = None
|
|
age_years: Optional[int] = Field(None, ge=0, le=200)
|
|
distillery: Optional[str] = Field(None, max_length=255)
|
|
country: Optional[str] = Field(None, max_length=255)
|
|
|
|
|
|
class DrinkResponse(DrinkBase):
|
|
model_config = ConfigDict(from_attributes=True)
|
|
|
|
id: int
|
|
owner_id: int
|
|
image_path: Optional[str] = None
|
|
created_at: datetime
|
|
updated_at: datetime
|