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

refactor

Browse Source
This commit is contained in:
Julius Unverfehrt 2022-06-24 12:35:29 +02:00
parent 80f04e5449
commit c7ce79ebbe
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/mock_client.py
View File

@ -48,11 +48,11 @@ def make_connection() -> pika.BlockingConnection:
return connection return connection
def build_message_bodies(analyse_container_type, bucket_name): def build_message_bodies(analysis_container, bucket_name):
def update_message(message_dict): def update_message(message_dict):
if analyse_container_type == "detr" or analyse_container_type == "image": if analysis_container == "detr" or analysis_container == "image":
message_dict.update({"targetFileExtension": "ORIGIN.pdf.gz", "responseFileExtension": "IMAGE_INFO.json.gz"}) message_dict.update({"targetFileExtension": "ORIGIN.pdf.gz", "responseFileExtension": "IMAGE_INFO.json.gz"})
if analyse_container_type == "conversion": if analysis_container == "conversion":
message_dict.update( message_dict.update(
{ {
"targetFileExtension": "ORIGIN.pdf.gz", "targetFileExtension": "ORIGIN.pdf.gz",
@ -61,20 +61,20 @@ def build_message_bodies(analyse_container_type, bucket_name):
"pages": [1, 2, 3], "pages": [1, 2, 3],
} }
) )
if analyse_container_type == "table_parsing": if analysis_container == "table_parsing":
message_dict.update( message_dict.update(
{ {
"operation": "table_parsing", "operation": "table_parsing",
"pages": [1, 2, 3], "pages": [1, 2, 3],
} }
) )
if analyse_container_type == "extraction": if analysis_container == "extraction":
message_dict.update( message_dict.update(
{"targetFileExtension": "ORIGIN.pdf.gz", "responseFileExtension": "json.gz", "operation": "extraction"} {"targetFileExtension": "ORIGIN.pdf.gz", "responseFileExtension": "json.gz", "operation": "extraction"}
) )
if analyse_container_type == "dl_error": if analysis_container == "dl_error":
message_dict.update({"targetFileExtension": "no_such_file", "responseFileExtension": "IMAGE_INFO.json.gz"}) message_dict.update({"targetFileExtension": "no_such_file", "responseFileExtension": "IMAGE_INFO.json.gz"})
if analyse_container_type == "ner": if analysis_container == "ner":
message_dict.update( message_dict.update(
{"targetFileExtension": "TEXT.json.gz", "responseFileExtension": "NER_ENTITIES.json.gz"} {"targetFileExtension": "TEXT.json.gz", "responseFileExtension": "NER_ENTITIES.json.gz"}
) )