diff --git a/pyinfra/queue/queue_manager.py b/pyinfra/queue/queue_manager.py index 6a0a2f2..f5ffb98 100644 --- a/pyinfra/queue/queue_manager.py +++ b/pyinfra/queue/queue_manager.py @@ -12,7 +12,7 @@ from pika.adapters.blocking_connection import BlockingChannel from pyinfra.config import Config from pyinfra.exception import ProcessingFailure from pyinfra.payload_processing.processor import PayloadProcessor -from pyinfra.utils.dict import save_project +from pyinfra.utils.dict import safe_project CONFIG = Config() @@ -191,7 +191,7 @@ class QueueManager: try: self.logger.debug("Processing (%s, %s, %s)", frame, properties, body) - filtered_message_headers = save_project(properties.headers, ["X-TENANT-ID"]) # TODO: parametrize key? + filtered_message_headers = safe_project(properties.headers, ["X-TENANT-ID"]) # TODO: parametrize key? message_body = {**json.loads(body), **filtered_message_headers} processing_result = process_message_body_and_await_result(message_body) diff --git a/pyinfra/utils/dict.py b/pyinfra/utils/dict.py index a220d5a..a732a6d 100644 --- a/pyinfra/utils/dict.py +++ b/pyinfra/utils/dict.py @@ -1,5 +1,5 @@ from funcy import project -def save_project(mapping, keys) -> dict: +def safe_project(mapping, keys) -> dict: return project(mapping, keys) if mapping else {}