flibusta-apps/telegram_files_cache_server
1
0
Fork 0
mirror of https://github.com/flibusta-apps/telegram_files_cache_server.git synced 2025-12-06 06:35:38 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add logger

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-05-20 22:36:20 +02:00
parent 12d3e55e1c
commit aa2dd01c82
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/core/taskiq_middlewares.py
View File

@@ -1,3 +1,4 @@
import logging
from typing import Any from typing import Any
from taskiq import SimpleRetryMiddleware from taskiq import SimpleRetryMiddleware
@@ -5,10 +6,13 @@ from taskiq.message import TaskiqMessage
from taskiq.result import TaskiqResult from taskiq.result import TaskiqResult
logger = logging.Logger("taskiq_middleware", logging.INFO)
class FastAPIREtryMiddleware(SimpleRetryMiddleware): class FastAPIREtryMiddleware(SimpleRetryMiddleware):
@staticmethod @staticmethod
def _is_need_to_remove(to_remove: list[Any], value: Any) -> bool: 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 return type(value) in to_remove
async def on_error( async def on_error(