Fixed flake.

Signed-off-by: Pavel Kirilin <win10@list.ru>

Author: s3rius

taskiq_nats/__init__.py

@@ -1,3 +1,10 @@
+"""
+NATS integration for taskiq.
+
+This package contains broker that
+uses NATS as a message queue.
+"""
+
 from taskiq_nats.broker import NatsBroker
 
 __all__ = ["NatsBroker"]

taskiq_nats/broker.py

@@ -11,6 +11,20 @@
 
 
 class NatsBroker(AsyncBroker):
+    """
+    NATS broker for taskiq.
+
+    By default this broker works
+    broadcasting message to all connected workers.
+
+    If you want to make it work as queue,
+    you need to supply name of the queue in
+    queue argument.
+
+    Docs about queue:
+    https://docs.nats.io/nats-concepts/core-nats/queue
+    """
+
     def __init__(  # noqa: WPS211 (too many args)
         self,
         servers: Union[str, list[str]],
@@ -28,23 +42,40 @@ def __init__(  # noqa: WPS211 (too many args)
         self.subject = subject
 
     async def startup(self) -> None:
+        """
+        Startup event handler.
+
+        It simply connects to NATS cluster.
+        """
         await super().startup()
         await self.client.connect(self.servers, **self.connection_kwargs)
 
     async def kick(self, message: BrokerMessage) -> None:
+        """
+        Send a message using NATS.
+
+        :param message: message to send.
+        """
         await self.client.publish(
             self.subject,
             payload=message.json().encode(),
             headers=message.labels,
         )
 
     async def listen(self) -> AsyncGenerator[BrokerMessage, None]:
+        """
+        Start listen to new messages.
+
+        :yield: incoming messages.
+        """
         subscribe = await self.client.subscribe(self.subject, queue=self.queue or "")
         async for message in subscribe.messages:
             try:
                 yield BrokerMessage.parse_raw(message.data)
             except ValueError:
-                logger.warning(f"Cannot parse message: {message.data.decode('utf-8')}")
+                data = message.data.decode("utf-8")
+                logger.warning(f"Cannot parse message: {data}")
 
     async def shutdown(self) -> None:
+        """Close connections to NATS."""
         await self.client.close()