from datetime import datetime
from typing import Optional
from pydantic import BaseModel
from app.models.stock_movement import MovementType
from app.schemas.product import Product


class StockMovementBase(BaseModel):
    product_id: int
    movement_type: MovementType
    quantity: int
    reference: Optional[str] = None
    notes: Optional[str] = None


class StockMovementCreate(StockMovementBase):
    pass


class StockMovement(StockMovementBase):
    id: int
    created_at: datetime
    product: Optional[Product] = None

    class Config:
        from_attributes = True