from sqlalchemy import Column, Integer, String
from sqlalchemy.dialects.postgresql import BIT
from . import Base
from typing import Any
from app.utils.utility import to_dict


class OnboardingProfile(Base):
    __tablename__ = "OnboardingProfiles"

    onboardingId = Column(Integer, primary_key=True)
    userId = Column(Integer)
    isWorkingWithAnyLoanOfficer = Column(BIT, default="0")
    address = Column(String)
    lenderId = Column(Integer)
    loanPurposeId = Column(Integer)
    employmentTypeId = Column(Integer)
    statusId = Column(Integer)
    pageNumber = Column(Integer)

    def dict(self) -> dict[str, Any]:
        return to_dict(self)