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

class TradingMonthCreate(CustomModel):
    group_id: int 
    store_a_trading_months: str 
    store_b_trading_months: str 
    rationale_for_store_a: str
    rationale_for_store_b: str

class TradingMonthResponce(CustomModel):
    trading_id : int 
    group_id: int 
    store_a_trading_months: str 
    store_b_trading_months: str 
    rationale_for_store_a: str
    rationale_for_store_b: str   
    
class TradingMonthUpdate(CustomModel):
    trading_id : Optional[int] = None 
    group_id: Optional[int] = None 
    store_a_trading_months: Optional[str] = None 
    store_b_trading_months: Optional[str] = None  
    rationale_for_store_a: Optional[str] = None 
    rationale_for_store_b: Optional[str] = None