More responses for docs

Author: soundstorm

app/api/v1/bankomat.py

@@ -457,7 +457,7 @@ def set_pin(
     return {"detail": "PIN updated"}
 
 
-@router.get("/statement/{target_slug}")
+@router.get("/statement/{target_slug}", response_class=Response)
 def get_statement(
     target_slug: str,
     from_year: int = Query(...),
@@ -522,7 +522,7 @@ def get_statement(
         raise HTTPException(500, f"PDF generation failed: {e}") from e
 
 
-@router.get("/statement-all")
+@router.get("/statement-all", response_class=Response)
 def get_statement_all(
     from_year: int = Query(...),
     from_month: int = Query(...),

app/api/v1/products.py

@@ -32,6 +32,7 @@
     ProductStocktaking,
     ProductUpdate,
     PurchaseBody,
+    PurchaseResponse,
 )
 
 router = APIRouter()
@@ -333,7 +334,7 @@ def delete_alias(
 
 # --- Purchase (checkout device) ---
 
-@router.post("/products/{ean}/purchase")
+@router.post("/products/{ean}/purchase", response_model=PurchaseResponse)
 def purchase_product(
     ean: str,
     body: PurchaseBody,
@@ -373,4 +374,4 @@ def purchase_product(
     ))
     db.commit()
 
-    return {"detail": "Purchase successful", "product": product.name, "new_balance": float(user.balance)}
+    return PurchaseResponse(detail="Purchase successful", product=product.name, new_balance=user.balance)

app/api/v1/rentals.py

@@ -11,6 +11,7 @@
 from app.schemas.common import MessageResponse
 from app.schemas.rental import (
     ActiveRentalResponse,
+    RentalAuthorizeResponse,
     RentalCatalogItem,
     RentalItemCreate,
     RentalItemResponse,
@@ -133,7 +134,7 @@ def item_status(
 
 # --- Rental operations ---
 
-@router.get("/authorize/{nfc_id}")
+@router.get("/authorize/{nfc_id}", response_model=RentalAuthorizeResponse)
 def authorize_renter(
     nfc_id: int,
     device: Machine = Depends(get_current_device),

app/schemas/product.py

@@ -80,9 +80,10 @@ class PurchaseBody(BaseModel):
     nfc_id: int
 
 
-class ProductPurchase(BaseModel):
-    nfc_id: int
-    ean: str
+class PurchaseResponse(BaseModel):
+    detail: str
+    product: str
+    new_balance: Decimal = Field(ge=0, examples=[Decimal("4.20")])
 
 
 class ProductPopularityResponse(BaseModel):

app/schemas/rental.py

@@ -77,3 +77,9 @@ class RentalCatalogItem(BaseModel):
     uhf_tid: str
     name: str
     is_rented: bool
+
+
+class RentalAuthorizeResponse(BaseModel):
+    authorized: bool
+    user_id: int
+    user_name: Optional[str]