from pydantic import BaseModel
from datetime import date

class ProductCreate(BaseModel):
    product_id: int
    name: str
    description: str | None = None
    category: str

class ProductResponse(ProductCreate):
    id: int

    class Config:
        from_attributes = True

class StorageLocationCreate(BaseModel):
    location_id: int
    name: str
    capacity: int = 0

class StorageLocationResponse(StorageLocationCreate):
    id: int

    class Config:
        from_attributes = True

class TransferCreate(BaseModel):
    transfer_id: int
    product_id: int
    from_location_id: int
    to_location_id: int
    quantity: int = 0

class TransferResponse(TransferCreate):
    id: int

    class Config:
        from_attributes = True