upd readme

Author: Friskes

README.md

@@ -69,6 +69,7 @@ def test_request():
 
 ```python
 # NOTE: The with context manager does not support multi-launch number_runs > 1
+# NOTE: Also you can use `async with` if you capture queries in async context.
 with CaptureQueries(number_runs=1, advanced_verb=True) as ctx:
     response = self.client.get(url)
 
@@ -123,4 +124,5 @@ settings.PRINTER_HANDLERS.append("path.to.your.handler.SomeHandler")
 ```
 
 ## TODO:
-1. Add support for other ORM's, SQLAlchemy, etc.
+1. Add support for async loop and async func decorator, __acall__, __aiter__, __anext__
+2. Add support for other ORM's, SQLAlchemy, etc.

src/capture_db_queries/decorators.py

@@ -179,16 +179,15 @@ def wrapper_cls(self) -> type[BaseExecutionWrapper]:
             return ExplainExecutionWrapper
         return BaseExecutionWrapper
 
-    async def __aenter__(self):
-        await sync_to_async(self.__enter__)()
-        return self
+    async def __aenter__(self) -> Self:
+        return await sync_to_async(self.__enter__)()
 
     async def __aexit__(
         self,
         exc_type: type[BaseException] | None,
         exc_value: BaseException | None,
         traceback: TracebackType | None,
-    ):
+    ) -> None:
         await sync_to_async(self.__exit__)(exc_type, exc_value, traceback)
 
     def __enter__(self) -> Self: