From b3a68a264a08af77fc047f97f9adb7453b77f037 Mon Sep 17 00:00:00 2001 From: Daniel Roth Date: Wed, 13 May 2026 13:32:54 +0000 Subject: [PATCH] =?UTF-8?q?new=20files=20types=20inferred=20from=20file=20?= =?UTF-8?q?names=20=F0=9F=9F=A9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- backend/pashub_fetcher/core_files.py | 10 ++++++++++ backend/pashub_fetcher/tests/test_core_files.py | 6 ++---- 2 files changed, 12 insertions(+), 4 deletions(-) diff --git a/backend/pashub_fetcher/core_files.py b/backend/pashub_fetcher/core_files.py index aa426475..b5ce1073 100644 --- a/backend/pashub_fetcher/core_files.py +++ b/backend/pashub_fetcher/core_files.py @@ -39,4 +39,14 @@ def infer_file_type(filename: str) -> Optional[str]: for core_file, file_type in CORE_TO_FILETYPE_MAP.items(): if filename.startswith(core_file.value): return file_type + + if CoreFiles.IMPROVEMENT_OPTION_EVALUATION.value in filename: + return CORE_TO_FILETYPE_MAP[CoreFiles.IMPROVEMENT_OPTION_EVALUATION] + + if CoreFiles.MEDIUM_TERM_IMPROVEMENT_PLAN.value in filename: + return CORE_TO_FILETYPE_MAP[CoreFiles.MEDIUM_TERM_IMPROVEMENT_PLAN] + + if "-OSM-" in filename and "DR-N-A" in filename: + return CORE_TO_FILETYPE_MAP[CoreFiles.RETROFIT_DESIGN_DOC] + return None diff --git a/backend/pashub_fetcher/tests/test_core_files.py b/backend/pashub_fetcher/tests/test_core_files.py index fca29b7e..f8e8b431 100644 --- a/backend/pashub_fetcher/tests/test_core_files.py +++ b/backend/pashub_fetcher/tests/test_core_files.py @@ -1,5 +1,3 @@ -import pytest - from backend.pashub_fetcher.core_files import infer_file_type @@ -59,6 +57,6 @@ def test_infer_medium_term_improvement_plan(): assert infer_file_type("60800802 - NG4 4HD - Medium Term Improvement Plan IOE.pdf") == "medium_term_improvement_plan" -@pytest.mark.skip(reason="Retrofit Design Doc filename pattern not yet known") def test_infer_retrofit_design_doc(): - assert infer_file_type("2512-OSM-H56M900-XX-DR-N-A_Radford Road 408.pdf") == "retrofit_design_doc" + assert infer_file_type("2512-OSM-H21M900-XX-DR-N-A_Lord Nelson Street 018.pdf") == "retrofit_design_doc" + assert infer_file_type("2603-OSM-B06M901-XX-DR-N-A_Alvaston Walk 022.pdf") == "retrofit_design_doc"