From aa2dd01c8215d392e96a8cf27242d2ea0fb4be8e Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Sat, 20 May 2023 22:36:20 +0200 Subject: [PATCH] Add logger --- src/core/taskiq_middlewares.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/core/taskiq_middlewares.py b/src/core/taskiq_middlewares.py index 80ac467..076bf06 100644 --- a/src/core/taskiq_middlewares.py +++ b/src/core/taskiq_middlewares.py @@ -1,3 +1,4 @@ +import logging from typing import Any from taskiq import SimpleRetryMiddleware @@ -5,10 +6,13 @@ from taskiq.message import TaskiqMessage from taskiq.result import TaskiqResult +logger = logging.Logger("taskiq_middleware", logging.INFO) + + class FastAPIREtryMiddleware(SimpleRetryMiddleware): @staticmethod def _is_need_to_remove(to_remove: list[Any], value: Any) -> bool: - print(type(value), to_remove) + logger.info(f"{type(value)}, {to_remove}") return type(value) in to_remove async def on_error(