From bd797e8e6c97b977f33f9676a6034b2bb8969de1 Mon Sep 17 00:00:00 2001 From: "d.tuytuyshkin@gmail.com" Date: Mon, 2 Aug 2021 14:56:00 +0300 Subject: [PATCH] Added sync support for DebugMiddleware --- panini/middleware/debug_middleware.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/panini/middleware/debug_middleware.py b/panini/middleware/debug_middleware.py index 3880119..2cdc7dd 100644 --- a/panini/middleware/debug_middleware.py +++ b/panini/middleware/debug_middleware.py @@ -1,3 +1,5 @@ +import asyncio + from panini.middleware import Middleware from panini.utils.logger import get_logger @@ -32,10 +34,16 @@ async def send_any(self, subject: str, message, send_func, *args, **kwargs): async def listen_any(self, msg, callback): if not self.use_listen_any: - return await callback(msg) + if asyncio.iscoroutinefunction(callback): + return await callback(msg) + else: + return callback(msg) self._log(f"Listen to {msg.subject} with payload {msg.data}") - response = await callback(msg) + if asyncio.iscoroutinefunction(callback): + response = await callback(msg) + else: + response = callback(msg) if response is not None: self._log(f"Answered to it: {response}")