fix formatting

Author: dadodimauro

rabbitmq_amqp_python_client/asyncio/connection.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import logging
 from typing import Callable, Optional, Union
@@ -52,14 +54,14 @@ def __init__(
         self._async_publishers: list[AsyncPublisher] = []
         self._async_consumers: list[AsyncConsumer] = []
         self._async_managements: list[AsyncManagement] = []
-        self._remove_callback = None
+        self._remove_callback: Optional[Callable[[AsyncConnection], None]] = None
 
     async def dial(self) -> None:
         async with self._connection_lock:
             await self._event_loop.run_in_executor(None, self._connection.dial)
 
     def _set_remove_callback(
-        self, callback: Optional[Callable[["AsyncConnection"], None]]
+        self, callback: Optional[Callable[[AsyncConnection], None]]
     ) -> None:
         self._remove_callback = callback
 
@@ -181,7 +183,7 @@ async def consumer(
             async_consumer = AsyncConsumer(
                 self._connection._conn,
                 destination,
-                message_handler,  # type: ignore
+                message_handler,  # pyright: ignore[reportArgumentType]
                 consumer_options,
                 credit,
                 loop=self._event_loop,
@@ -202,7 +204,7 @@ async def refresh_token(self, token: str) -> None:
                 token,
             )
 
-    async def __aenter__(self) -> "AsyncConnection":
+    async def __aenter__(self) -> AsyncConnection:
         await self.dial()
         return self
 

rabbitmq_amqp_python_client/asyncio/consumer.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import logging
 from types import TracebackType
@@ -31,7 +33,7 @@ def __init__(
         self._connection_lock = connection_lock or asyncio.Lock()
 
     def _set_remove_callback(
-        self, callback: Optional[Callable[["AsyncConsumer"], None]]
+        self, callback: Optional[Callable[[AsyncConsumer], None]]
     ) -> None:
         self._remove_callback = callback
 
@@ -62,7 +64,7 @@ async def stop(self) -> None:
         async with self._connection_lock:
             await self._event_loop.run_in_executor(None, self._consumer.stop)
 
-    async def __aenter__(self) -> "AsyncConsumer":
+    async def __aenter__(self) -> AsyncConsumer:
         return self
 
     async def __aexit__(

rabbitmq_amqp_python_client/asyncio/enviroment.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import logging
 from typing import Optional, Union
@@ -86,7 +88,7 @@ async def close(self) -> None:
     async def connections(self) -> list[AsyncConnection]:
         return self._connections
 
-    async def __aenter__(self) -> "AsyncEnvironment":
+    async def __aenter__(self) -> AsyncEnvironment:
         return self
 
     async def __aexit__(

rabbitmq_amqp_python_client/asyncio/management.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import asyncio
 import logging
 from typing import Any, Callable, Optional, Union
@@ -36,7 +38,7 @@ def __init__(
         self._connection_lock = connection_lock or asyncio.Lock()
 
     def _set_remove_callback(
-        self, callback: Optional[Callable[["AsyncManagement"], None]]
+        self, callback: Optional[Callable[[AsyncManagement], None]]
     ) -> None:
         self._remove_callback = callback
 
@@ -168,7 +170,7 @@ async def refresh_token(self, token: str) -> None:
                 token,
             )
 
-    async def __aenter__(self) -> "AsyncManagement":
+    async def __aenter__(self) -> AsyncManagement:
         await self.open()
         return self
 

rabbitmq_amqp_python_client/asyncio/publisher.py

@@ -25,7 +25,7 @@ def __init__(
         self._publisher = Publisher(conn, addr)
         self._loop = loop
         self._connection_lock = connection_lock or asyncio.Lock()
-        self._remove_callback = None
+        self._remove_callback: Optional[Callable[["AsyncPublisher"], None]] = None
 
     def _set_remove_callback(
         self, callback: Optional[Callable[["AsyncPublisher"], None]]
@@ -41,7 +41,7 @@ async def publish(self, message: Message) -> Delivery:
     async def close(self) -> None:
         if not self.is_open:
             return
-    
+
         try:
             async with self._connection_lock:
                 await self._event_loop.run_in_executor(None, self._publisher.close)
@@ -69,7 +69,6 @@ def _event_loop(self) -> asyncio.AbstractEventLoop:
 
     @property
     def is_open(self) -> bool:
-        print( self._publisher.is_open )
         return self._publisher.is_open
 
     @property