Push V1 app

backend/schemas.py

@@ -0,0 +1,162 @@
+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