Spaces:
Sleeping
Sleeping
| import abc | |
| from enums.enums import Questions, DetailsType | |
| class SavedDetails(abc.ABC): | |
| excluded_fields = ["type_"] | |
| def __init__(self, type_: DetailsType, key: str = None): | |
| self.type_ = type_ | |
| self.key = key | |
| def load(cls, data: dict): | |
| type_ = data.get("type_") | |
| if not type_ or type_ != cls.type_: | |
| raise ValueError(f"the expected type is {cls.type_} but is actually {type_}") | |
| return cls.__init__(**{k: v for k, v in data if k != "type"}) | |
| def to_json(self): | |
| self.key = self.key.replace(" ", "_") | |
| return {k: v for k, v in self.__dict__.items() if k not in self.excluded_fields} | |
| def to_answers(self): | |
| pass | |
| def short_description(self): | |
| return ", ".join([v for k, v in self.__dict__.items() if k not in self.excluded_fields]) | |
| def widget_labels(self) -> dict[str, str]: | |
| fields = {k: " ".join(k.title().split("_")) for k in self.__dict__ | |
| if k not in self.excluded_fields and k != "key"} | |
| fields["key"] = "Save as" | |
| return fields | |
| class PersonalDetails(SavedDetails): | |
| def __init__(self, key: str = None, full_name: str = None, email: str = None, contact_number: str = None, *_): | |
| super().__init__(DetailsType.PERSONAL_DETAILS, key) | |
| self.full_name = full_name | |
| self.email = email | |
| self.contact_number = contact_number | |
| def to_answers(self) -> dict[Questions, str]: | |
| answers = {} | |
| if self.full_name is not None: | |
| answers[Questions.FULL_NAME] = self.full_name | |
| if self.email is not None: | |
| answers[Questions.YOUR_EMAIL] = self.email | |
| if self.contact_number is not None: | |
| answers[Questions.CONTACT_NUMBER] = self.contact_number | |
| return answers | |
| class LocationDetails(SavedDetails): | |
| type_ = DetailsType.LOCATION_DETAILS | |
| def __init__(self, owner_or_tenant: str = None, community: str = None, building: str = None, | |
| unit_number: str = None, key: str = None, *_): | |
| super().__init__(self.type_, key) | |
| self.owner_or_tenant = owner_or_tenant | |
| self.community = community | |
| self.building = building | |
| self.unit_number = unit_number | |
| def to_answers(self) -> dict[Questions, str]: | |
| answers = {} | |
| if self.owner_or_tenant is not None: | |
| answers[Questions.OWNER_OR_TENANT] = self.owner_or_tenant | |
| if self.building is not None: | |
| answers[Questions.BUILDING] = self.building | |
| if self.community is not None: | |
| answers[Questions.COMMUNITY] = self.community | |
| if self.unit_number is not None: | |
| answers[Questions.UNIT_APT_NUMBER] = self.unit_number | |
| return answers | |
| class ContractorDetails(SavedDetails): | |
| type_ = DetailsType.CONTRACTOR_DETAILS | |
| def __init__(self, contractor_name: str = None, contractor_contact_number: str = None, contractor_email: str = None, | |
| key: str = None, *_): | |
| super().__init__(self.type_, key) | |
| self.contractor_name = contractor_name | |
| self.contractor_contact_number = contractor_contact_number | |
| self.contractor_email = contractor_email | |
| def to_answers(self) -> dict[Questions, str]: | |
| answers = {} | |
| if self.contractor_email is not None: | |
| answers[Questions.COMPANY_EMAIL] = self.contractor_email | |
| if self.contractor_contact_number is not None: | |
| answers[Questions.COMPANY_NUMBER] = self.contractor_contact_number | |
| if self.contractor_name is not None: | |
| answers[Questions.COMPANY_NAME] = self.contractor_name | |
| return answers | |
| detail_classes = { | |
| DetailsType.LOCATION_DETAILS: LocationDetails, | |
| DetailsType.PERSONAL_DETAILS: PersonalDetails, | |
| DetailsType.CONTRACTOR_DETAILS: ContractorDetails | |
| } | |