From 2bc520c849ea4e833cb60b2c97626da6636d3155 Mon Sep 17 00:00:00 2001
From: Julius Unverfehrt <julius.unverfehrt@iqser.com>
Date: Tue, 22 Mar 2022 15:42:08 +0100
Subject: [PATCH] adjust mock script

---
 scripts/mock_client.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/scripts/mock_client.py b/scripts/mock_client.py
index 89b2b90..e7699fa 100644
--- a/scripts/mock_client.py
+++ b/scripts/mock_client.py
@@ -54,6 +54,8 @@ def build_message_bodies(analyse_container_type):
 
     storage = get_s3_storage()
     for bucket_name, pdf_name in storage.get_all_object_names(CONFIG.storage.bucket):
+        if "pdf" not in pdf_name:
+            continue
         file_id = pdf_name.split(".")[0]
         dossier_id, file_id = file_id.split("/")
         message_dict = {"dossierId": dossier_id, "fileId": file_id}
@@ -74,6 +76,7 @@ def main(args):
     for method_frame, _, body in channel.consume(queue=CONFIG.rabbitmq.queues.output):
         print(f"Received {json.loads(body)}")
         channel.basic_ack(method_frame.delivery_tag)
+    channel.close()
 
 
 if __name__ == "__main__":