dam_helper/enums/enums.py

from enum import Enum


class Questions(Enum):
    FULL_NAME = 0
    WORK_TO_DO = 1
    COMMUNITY = 2
    BUILDING = 3
    UNIT_APT_NUMBER = 4
    OWNER_OR_TENANT = 5
    START_DATE = 6
    END_DATE = 7
    CONTACT_NUMBER = 8
    COMPANY_NAME = 9
    COMPANY_EMAIL = 10
    COMPANY_NUMBER = 11
    YOUR_EMAIL = 12

    @classmethod
    def values(cls):
        return [Questions.FULL_NAME, Questions.WORK_TO_DO, Questions.BUILDING, Questions.UNIT_APT_NUMBER,
                Questions.OWNER_OR_TENANT, Questions.START_DATE, Questions.END_DATE, Questions.CONTACT_NUMBER,
                Questions.COMPANY_NAME, Questions.COMPANY_EMAIL, Questions.COMPANY_NUMBER, Questions.YOUR_EMAIL]


class DetailsType(Enum):
    PERSONAL_DETAILS = 1
    LOCATION_DETAILS = 2
    CONTRACTOR_DETAILS = 3

    def title(self):
        return self.name.replace("_", " ").lower()

    @classmethod
    def values(cls):
        return [cls.PERSONAL_DETAILS, cls.LOCATION_DETAILS, cls.CONTRACTOR_DETAILS]


class Steps(Enum):
    INITIAL_STATE = 1
    PARSING_ANSWERS = 2
    ASK_AGAIN = 3
    FIND_CATEGORIES = 4
    VALIDATE_DATA = 5
    PARSING_ERROR = 6
    FORM_CREATED = 7