refactor lint

Author: vladimirgubarik

app/decorators/mapper/exception_mapper.py

@@ -1,5 +1,6 @@
 """Decorators to rethrow and log exceptions."""
 
+import asyncio
 from functools import wraps
 from inspect import iscoroutinefunction
 from typing import Any, Callable, Type
@@ -81,6 +82,9 @@ def _handle_exception_logic(
         args: tuple[Any, ...],
         kwargs: dict[str, Any],
     ) -> None:
+        if isinstance(exc, asyncio.CancelledError):
+            raise
+
         if exception_factory := self._get_exception_factory(type(exc)):
             context = ExceptionContext(exc, func, self._filtered_args(args), kwargs)
             raise exception_factory.make_exception(context) from exc

app/infrastructure/containers.py

@@ -1,7 +1,7 @@
 import asyncio
 
 from dependency_injector import containers, providers
-from dependency_injector.providers import Callable, Factory, Singleton
+from dependency_injector.providers import Callable, Factory
 from httpx import AsyncClient, Limits
 from pybotx import Bot, HandlerCollector
 from redis import asyncio as aioredis

app/infrastructure/repositories/sample_record.py

@@ -72,7 +72,6 @@ async def create(self, record: SampleRecord) -> SampleRecord:
             .returning(SampleRecordModel)
         )
         result = await self._session.execute(query)
-        await self._session.flush()
         record_model = result.scalar_one()
         return self._to_domain_object(record_model)
 
@@ -91,7 +90,6 @@ async def update(self, record: SampleRecord) -> SampleRecord:
             .returning(SampleRecordModel)
         )
         execute_result = (await self._session.execute(query)).scalar_one_or_none()
-        await self._session.flush()
         if execute_result is None:
             raise RecordDoesNotExistError(
                 f"Sample record with id={record.id} does not exist."
@@ -125,8 +123,6 @@ async def delete(self, record_id: int) -> None:
                 f"Sample record with id={record_id} does not exist."
             )
 
-        await self._session.flush()
-
     @ExceptionMapper(
         {
             NoResultFound: EnrichedExceptionFactory(RecordDoesNotExistError),

app/infrastructure/worker/worker.py

@@ -7,7 +7,7 @@
 from redis.asyncio import Redis
 from saq import CronJob, Queue
 
-from app.infrastructure.containers import WorkerStartupContainer, BaseStartupContainer
+from app.infrastructure.containers import BaseStartupContainer, WorkerStartupContainer
 from app.infrastructure.worker.tasks.simple_task import heartbeat_task
 from app.logger import logger
 from app.settings import settings

app/presentation/bot/command_handlers/sample_record.py

@@ -17,8 +17,8 @@
     SendErrorExplainToUserHandler,
 )
 from app.presentation.bot.error_handlers.exceptions_chain_executor import (
-    DEFAULT_HANDLERS,
-    ExceptionHandlersChainExecutor, DEFAULT_HANDLERS_WITH_EXPLAIN,
+    DEFAULT_HANDLERS_WITH_EXPLAIN,
+    ExceptionHandlersChainExecutor,
 )
 from app.presentation.bot.resources.strings import (
     SAMPLE_RECORD_CREATED_ANSWER,
@@ -48,9 +48,6 @@ class CreateSampleRecordHandler(BaseCommandHandler):
             }
         )
     ]
-    exception_handler_chain_executor = ExceptionHandlersChainExecutor(
-        _EXCEPTIONS_HANDLERS
-    )
 
     def __init__(
         self,
@@ -62,7 +59,11 @@ def __init__(
         self._use_cases = use_case_factory
         self.unit_of_work = unit_of_work
 
-        super().__init__(bot, message, self.exception_handler_chain_executor)
+        exception_handler_chain_executor = ExceptionHandlersChainExecutor(
+            self._EXCEPTIONS_HANDLERS
+        )
+
+        super().__init__(bot, message, exception_handler_chain_executor)
 
     async def handle_logic(
         self,
@@ -94,9 +95,6 @@ class DeleteSampleRecordHandler(BaseCommandHandler):
             }
         )
     ]
-    exception_handler_chain_executor = ExceptionHandlersChainExecutor(
-        _EXCEPTIONS_HANDLERS
-    )
 
     def __init__(
         self,
@@ -107,8 +105,11 @@ def __init__(
     ):
         self._use_cases = use_case_factory
         self.unit_of_work = unit_of_work
+        exception_handler_chain_executor = ExceptionHandlersChainExecutor(
+            self._EXCEPTIONS_HANDLERS
+        )
 
-        super().__init__(bot, message, self.exception_handler_chain_executor)
+        super().__init__(bot, message, exception_handler_chain_executor)
 
     async def handle_logic(
         self,
@@ -153,7 +154,11 @@ def __init__(
         self._use_cases = use_case_factory
         self.unit_of_work = unit_of_work
 
-        super().__init__(bot, message, self.exception_handler_chain_executor)
+        exception_handler_chain_executor = ExceptionHandlersChainExecutor(
+            self._EXCEPTIONS_HANDLERS
+        )
+
+        super().__init__(bot, message, exception_handler_chain_executor)
 
     async def handle_logic(
         self,