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

class SegmentCreate(CustomModel):
    name: Optional[str] = None
    location: Optional[str] = None
    chief_wage_earner_occupation: Optional[str] = None
    highest_educated_male: Optional[str] = None
    highest_educated_female: Optional[str] = None
    ISEC_Segment: Optional[int] = None
    total_population : Optional[int] = None
    additional_information : Optional[str] = None
    other_details : Optional[str] = None
    group_id: int
    

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)