from pydantic import Field
from app.utils.schemas_utils import CustomModel

class customer_location_creat(CustomModel):
    
    ISEC_Segment: int
    sample_size: int 
    average_age: int 
    income_levels: str 
    occupation_mode: str
    education_level_mode: str 
    shopping_frequency_per_month: int
    household_consumption_per_month: int 
    group_id : int

class TotalConsumptionResponse(CustomModel):
    group_id: int
    total_household_consumption_per_month: int | None=Field(default=None)

class customerLocation(customer_location_creat):
    pass

class customerLocationResponse(CustomModel):
    
    ISEC_Segment: int 
    sample_size: int 
    average_age: int 
    income_levels: str 
    occupation_mode: str
    education_level_mode: str 
    shopping_frequency_per_month: int
    household_consumption_per_month: int 
    group_id : int

class customerLocationUpdate(CustomModel):

    customer_location_id: int 
    ISEC_Segment: int | None = Field(default=None)
    sample_size: int | None = Field(default=None)
    average_age: int | None = Field(default=None)
    income_levels: str | None = Field(default=None)
    occupation_mode: str| None = Field(default=None)
    education_level_mode: str | None = Field(default=None)
    shopping_frequency_per_month: int| None = Field(default=None)
    household_consumption_per_month: int | None = Field(default=None)
    group_id : int | None = Field(default=None)