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

class SegmentCreate(CustomModel):
    name: str | None = Field(default=None)
    location: str | None = Field(default=None)
    chief_wage_earner_occupation: str | None = Field(default=None)
    highest_educated_male: str | None = Field(default=None)
    highest_educated_female: str | None = Field(default=None)
    ISEC_Segment: int | None = Field(default=None)
    #catchment_size: int | None = Field(default=None)
    total_population : int | None = Field(default=None)
    additional_information : str | None = Field(default=None)
    other_details : str | None = Field(default=None)
    group_id: int | None = Field(default=None)
    

class SegmentUpdate(SegmentCreate):
    segment_id: int

class SegmantResponse(SegmentCreate):
    segment_id: int    
    
class OneField(CustomModel):
    segment_id: int | None = Field(default=None)
    ISEC_Segment: int | None = Field(default=None)
    group_id: int | None = Field(default=None)