From 187055e5ebcd3ffd9fd53c394ab9aca68ae39425 Mon Sep 17 00:00:00 2001 From: Julius Unverfehrt Date: Tue, 31 May 2022 18:19:41 +0200 Subject: [PATCH] adjust mock_client script for conversion --- scripts/mock_client.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/scripts/mock_client.py b/scripts/mock_client.py index e8ad0fa..8a2624c 100644 --- a/scripts/mock_client.py +++ b/scripts/mock_client.py @@ -10,7 +10,7 @@ from pyinfra.storage.storages import get_s3_storage def parse_args(): parser = argparse.ArgumentParser() parser.add_argument("--bucket_name", "-b", required=True) - parser.add_argument("--analysis_container", "-a", choices=["detr", "ner", "image"], required=True) + parser.add_argument("--analysis_container", "-a", choices=["detr", "ner", "image", "conversion"], required=True) args = parser.parse_args() return args @@ -47,6 +47,8 @@ def build_message_bodies(analyse_container_type, bucket_name): def update_message(message_dict): if analyse_container_type == "detr" or analyse_container_type == "image": message_dict.update({"targetFileExtension": "ORIGIN.pdf.gz", "responseFileExtension": "IMAGE_INFO.json.gz"}) + if analyse_container_type == "conversion": + message_dict.update({"targetFileExtension": "ORIGIN.pdf.gz", "responseFileExtension": "png.gz", "operation": "conversion"}) if analyse_container_type == "ner": message_dict.update( {"targetFileExtension": "TEXT.json.gz", "responseFileExtension": "NER_ENTITIES.json.gz"}