Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

maybe dirty inter-com working

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-02-17 09:31:36 +01:00
parent 2917fb940e
commit 9a32fc4e9d
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/serve.py
View File

@ -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