from pydantic import BaseModel


class OnboardingProfileSchema(BaseModel):
    isWorkingWithAnyLoanOfficer: bool
    address: str
    lenderId: int
    loanPurposeId: int
    employmentTypeId: int
    statusId: int
    pageNumber: int


class OnboardingProfileCreate(OnboardingProfileSchema):
    address: str | None = None
    lenderId: int | None = None
    loanPurposeId: int | None = None
    employmentTypeId: int | None = None
    statusId: int | None = None
    pass


class OnboardingProfileUpdate(OnboardingProfileSchema):
    isWorkingWithAnyLoanOfficer: bool | None = None
    address: str | None = None
    lenderId: int | None = None
    loanPurposeId: int | None = None
    employmentTypeId: int | None = None
    statusId: int | None = None


class OnboardingProfileInDB(OnboardingProfileSchema):
    OnboardingId: int | None = None

    class Config:
        from_attributes = True