geMoldInsight/src/api/inventory/schemas/inventory_schemas.py

from pydantic import BaseModel
from typing import Optional


class InventoryResponse(BaseModel):
    id: int
    product_id: int
    product_name: str
    product_sku: str
    warehouse_id: int
    warehouse_name: str
    quantity: int
    locked_quantity: int
    available_quantity: int

    class Config:
        from_attributes = True


class InventoryCreate(BaseModel):
    product_id: int
    warehouse_id: int
    quantity: int = 0
    locked_quantity: int = 0
    batch_number: Optional[str] = None
    location: Optional[str] = None


class InventoryUpdate(BaseModel):
    quantity: Optional[int] = None
    locked_quantity: Optional[int] = None
    batch_number: Optional[str] = None
    location: Optional[str] = None