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 14:45:36 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-05-21 01:21:34 +02:00
parent 279c06b13a
commit 4fc6f53fee
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/core/taskiq_middlewares.py
View File

@@ -12,14 +12,11 @@ logger = logging.getLogger("taskiq_middleware")
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:
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(
self, message: TaskiqMessage, result: TaskiqResult[Any], exception: Exception self, message: TaskiqMessage, result: TaskiqResult[Any], exception: Exception
) -> None: ) -> None:
logger.info(f"{self.broker.custom_dependency_context}")
types_to_remove = list(self.broker.custom_dependency_context.keys()) types_to_remove = list(self.broker.custom_dependency_context.keys())
message.args = [ message.args = [
@@ -33,4 +30,6 @@ class FastAPIREtryMiddleware(SimpleRetryMiddleware):
if not self._is_need_to_remove(types_to_remove, value) if not self._is_need_to_remove(types_to_remove, value)
} }
raise Exception(f"{self.broker.custom_dependency_context=} {message=}")
return await super().on_error(message, result, exception) return await super().on_error(message, result, exception)