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


class ElectricalCablingCreate(CustomModel):
    # electrical_id : int
    feature_type : str
    number_tonnes : int
    havs_rate_unit : float 
    cost_unit : int
    lighting_type : str 
    rate_unit : int
    remark : str
    store_format_type : str 
    choice : str
    group_id : int
    
class ElectricalCablingResponse(ElectricalCablingCreate):
    electrical_id : int
    total : Optional[float] = None
    
class ElectricalCablingUpdate(CustomModel):
    electrical_id : Optional[int] = None
    feature_type : Optional[str] = None
    number_tonnes : Optional[int] = None
    havs_rate_unit : Optional[float] = None 
    cost_unit : Optional[int] = None
    lighting_type : Optional[str] = None 
    rate_unit : Optional[int] = None
    remark : Optional[str] = None
    store_format_type : Optional[str] = None 
    choice : Optional[str] = None
    group_id : Optional[int] = None

    
class ElectricalCablingSum(CustomModel):
    group_id: int
    store_format_type: str
    total_sum: Optional[int] = None