From 9a32fc4e9dbde327e3e9e926f10b4aca4e0fd974 Mon Sep 17 00:00:00 2001 From: Julius Unverfehrt Date: Thu, 17 Feb 2022 09:31:36 +0100 Subject: [PATCH] maybe dirty inter-com working --- src/serve.py | 18 +++++++++--------- 1 file changed, 9 insertions(+), 9 deletions(-) diff --git a/src/serve.py b/src/serve.py index 648ab90..a168ca8 100644 --- a/src/serve.py +++ b/src/serve.py @@ -18,15 +18,14 @@ from pyinfra.utils.file import dossier_id_and_file_id_to_compressed_storage_pdf_ def make_file_getter(storage): - def get_file(payload): + def get_file(payload, pdf_dir): with tempfile.TemporaryDirectory() as pdf_compressed_dir: - with tempfile.TemporaryDirectory() as pdf_dir: - dossier_id, file_id = itemgetter("dossierId", "fileId")(payload) - object_name = dossier_id_and_file_id_to_compressed_storage_pdf_object_name(dossier_id, file_id) - downloaded_file_path = download(storage, object_name, pdf_compressed_dir) - unzipped_file_path = unzip(downloaded_file_path, pdf_dir) + dossier_id, file_id = itemgetter("dossierId", "fileId")(payload) + object_name = dossier_id_and_file_id_to_compressed_storage_pdf_object_name(dossier_id, file_id) + downloaded_file_path = download(storage, object_name, pdf_compressed_dir) + unzipped_file_path = unzip(downloaded_file_path, pdf_dir) - return unzipped_file_path + return unzipped_file_path return get_file @@ -48,8 +47,9 @@ def make_payload_processor(analysis_endpoint): def process(payload: dict): logging.info(f"Processing {payload}...") - file_path = get_file(payload) - predictions = analyze_file(file_path) + with tempfile.TemporaryDirectory() as pdf_dir: + file_path = get_file(payload, pdf_dir) + predictions = analyze_file(file_path) return predictions return process