from typing import Literal, Optional
from app.utils.schemas_utils import CustomModel


class StoreLevelStaffNeedsCreate(CustomModel):
    # store_id : int
    designation : str
    average_salary : float
    count : int
    store_formate_type : Literal["Store Format A", "Store Format B"] 
    group_id : int
    
class StoreLevelStaffNeedsResponse(StoreLevelStaffNeedsCreate):
    store_id : int
    
class StoreLevelStaffNeedsUpdate(CustomModel):
    store_id : Optional[int] = None
    designation : Optional[str] = None
    average_salary : Optional[float] = None
    count : Optional[int] = None
    store_formate_type : Optional[Literal["Store Format A", "Store Format B"]] 
    group_id : Optional[int] = None