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