From 8819a9a3c365b104743a8e3d715ea1255c5eb0db Mon Sep 17 00:00:00 2001 From: Gijs Vermeulen Date: Thu, 1 Feb 2024 14:47:17 +0100 Subject: [PATCH] Pass correct arguments to wrapped function --- pyproject.toml | 2 +- src/amqpstorm_flask/RabbitMQ.py | 14 ++++++-------- 2 files changed, 7 insertions(+), 9 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 9160c40..d04b8cb 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,7 +6,7 @@ build-backend = "setuptools.build_meta" [project] name = "amqpstorm-flask" -version = "0.1.7" +version = "0.1.8" description = "amqpstorm library for Flask" readme = "README.md" authors = [{ name = "Inuits", email = "developers@inuits.eu" }] diff --git a/src/amqpstorm_flask/RabbitMQ.py b/src/amqpstorm_flask/RabbitMQ.py index c535349..0b3e0ad 100644 --- a/src/amqpstorm_flask/RabbitMQ.py +++ b/src/amqpstorm_flask/RabbitMQ.py @@ -157,14 +157,12 @@ def _publish_to_channel( properties=properties, ) - def __create_wrapper_function(self, f): - def wrapper_function( - message, consumer_tag, channel, body, envelope, properties - ): + def __create_wrapper_function(self, routing_key, f): + def wrapper_function(message): f( - routing_key=envelope.routing_key, - body=body, - message_id=properties.message_id, + routing_key=routing_key, + body=message.body, + message_id=message.message_id, ) return wrapper_function @@ -207,7 +205,7 @@ def new_consumer(): arguments=queue_arguments, ) self.channel.basic.qos(prefetch_count=1) - wrapped_f = self.__create_wrapper_function(f) + wrapped_f = self.__create_wrapper_function(routing_key, f) self.channel.basic.consume( wrapped_f, queue=queue_name, no_ack=self.queue_params.no_ack )