diff --git a/backend/app/db/models/uploaded_file.py b/backend/app/db/models/uploaded_file.py index c629f574..f3cfee79 100644 --- a/backend/app/db/models/uploaded_file.py +++ b/backend/app/db/models/uploaded_file.py @@ -18,6 +18,9 @@ class FileTypeEnum(enum.Enum): ECMK_RD_SAP_SITE_NOTE = "ecmk_rd_sap_site_note" ECMK_SURVEY_XML = "ecmk_survey_xml" MAGIC_PLAN_JSON = "magic_plan_json" + IMPROVEMENT_OPTION_EVALUATION = "improvement_option_evaluation" + MEDIUM_TERM_IMPROVEMENT_PLAN = "medium_term_improvement_plan" + RETROFIT_DESIGN_DOC = "retrofit_design_doc" class FileSourceEnum(enum.Enum): diff --git a/backend/pashub_fetcher/core_files.py b/backend/pashub_fetcher/core_files.py index 4da10661..aa426475 100644 --- a/backend/pashub_fetcher/core_files.py +++ b/backend/pashub_fetcher/core_files.py @@ -14,6 +14,9 @@ class CoreFiles(Enum): PAR_PHOTOPACK = "PAR Photo Pack" PAS2023_PROPERTY = "PAS 2023 Property Assessment Report" PAS2023_OCCUPANCY = "PAS 2023 Occupancy Assessment Report" + IMPROVEMENT_OPTION_EVALUATION = "Improvement Option Evaluation" + MEDIUM_TERM_IMPROVEMENT_PLAN = "Medium Term Improvement Plan" + RETROFIT_DESIGN_DOC = "Retrofit Design Doc" CORE_TO_FILETYPE_MAP = { @@ -26,6 +29,9 @@ CORE_TO_FILETYPE_MAP = { CoreFiles.PAR_PHOTOPACK: FileTypeEnum.PAR_PHOTO_PACK.value, CoreFiles.PAS2023_PROPERTY: FileTypeEnum.PAS_2023_PROPERTY.value, CoreFiles.PAS2023_OCCUPANCY: FileTypeEnum.PAS_2023_OCCUPANCY.value, + CoreFiles.IMPROVEMENT_OPTION_EVALUATION: FileTypeEnum.IMPROVEMENT_OPTION_EVALUATION.value, + CoreFiles.MEDIUM_TERM_IMPROVEMENT_PLAN: FileTypeEnum.MEDIUM_TERM_IMPROVEMENT_PLAN.value, + CoreFiles.RETROFIT_DESIGN_DOC: FileTypeEnum.RETROFIT_DESIGN_DOC.value, }