ci: Add and configure PyLint, and make it happy

ssb/packet_stream.py

@@ -1,3 +1,5 @@
+"""Packet streams"""
+
 import logging
 import struct
 from asyncio import Event, Queue
@@ -7,28 +9,34 @@ from math import ceil
 
 import simplejson
 
-from secret_handshake import SHSClient, SHSServer
-
 
 logger = logging.getLogger("packet_stream")
 
 
 class PSMessageType(Enum):
+    """Available message types"""
+
     BUFFER = 0
     TEXT = 1
     JSON = 2
 
 
-class PSStreamHandler(object):
+class PSStreamHandler:
+    """Packet stream handler"""
+
     def __init__(self, req):
         super(PSStreamHandler).__init__()
         self.req = req
         self.queue = Queue()
 
     async def process(self, msg):
+        """Process a pending message"""
+
         await self.queue.put(msg)
 
     async def stop(self):
+        """Stop a pending request"""
+
         await self.queue.put(None)
 
     def __aiter__(self):
@@ -43,30 +51,39 @@ class PSStreamHandler(object):
         return elem
 
 
-class PSRequestHandler(object):
+class PSRequestHandler:
+    """Packet stream request handler"""
+
     def __init__(self, req):
-        super(PSRequestHandler).__init__()
         self.req = req
         self.event = Event()
         self._msg = None
 
     async def process(self, msg):
+        """Process a message request"""
+
         self._msg = msg
         self.event.set()
 
     async def stop(self):
+        """Stop a pending event request"""
+
         if not self.event.is_set():
             self.event.set()
 
-    def __await__(self):
+    async def __await__(self):
         # wait until 'process' is called
-        yield from self.event.wait().__await__()
+        await self.event.wait()
+
         return self._msg
 
 
-class PSMessage(object):
+class PSMessage:
+    """Packet Stream message"""
+
     @classmethod
     def from_header_body(cls, flags, req, body):
+        """Parse a raw message"""
         type_ = PSMessageType(flags & 0x03)
 
         if type_ == PSMessageType.TEXT:
@@ -78,13 +95,17 @@ class PSMessage(object):
 
     @property
     def data(self):
+        """The raw message data"""
+
         if self.type == PSMessageType.TEXT:
             return self.body.encode("utf-8")
-        elif self.type == PSMessageType.JSON:
+
+        if self.type == PSMessageType.JSON:
             return simplejson.dumps(self.body).encode("utf-8")
+
         return self.body
 
-    def __init__(self, type_, body, stream, end_err, req=None):
+    def __init__(self, type_, body, stream, end_err, req=None):  # pylint: disable=too-many-arguments
         self.stream = stream
         self.end_err = end_err
         self.type = type_
@@ -93,29 +114,35 @@ class PSMessage(object):
 
     def __repr__(self):
         if self.type == PSMessageType.BUFFER:
-            body = "{} bytes".format(len(self.body))
+            body = f"{len(self.body)} bytes"
         else:
             body = self.body
-        return "<PSMessage ({}): {}{} {}{}>".format(
-            self.type.name,
-            body,
-            "" if self.req is None else " [{}]".format(self.req),
-            "~" if self.stream else "",
-            "!" if self.end_err else "",
-        )
+
+        req = "" if self.req is None else f" [{self.req}]"
+        is_stream = "~" if self.stream else ""
+        err = "!" if self.end_err else ""
+
+        return f"<PSMessage ({self.type.name}): {body}{req} {is_stream}{err}>"
 
 
-class PacketStream(object):
+class PacketStream:
+    """SSB Packet stream"""
+
     def __init__(self, connection):
         self.connection = connection
         self.req_counter = 1
         self._event_map = {}
+        self._connected = False
 
     def register_handler(self, handler):
+        """Register an RPC handler"""
+
         self._event_map[handler.req] = (time(), handler)
 
     @property
     def is_connected(self):
+        """Check if the stream is connected"""
+
         return self.connection.is_connected
 
     def __aiter__(self):
@@ -148,7 +175,8 @@ class PacketStream(object):
             n_packets = ceil(length / 4096)
 
             body = b""
-            for n in range(n_packets):
+
+            for _ in range(n_packets):
                 body += await self.connection.read()
 
             logger.debug("READ %s %s", header, len(body))
@@ -159,12 +187,14 @@ class PacketStream(object):
             return None
 
     async def read(self):
+        """Read data from the packet stream"""
+
         msg = await self._read()
         if not msg:
             return None
         # check whether it's a reply and handle accordingly
         if msg.req < 0:
-            t, handler = self._event_map[-msg.req]
+            _, handler = self._event_map[-msg.req]
             await handler.process(msg)
             logger.info("RESPONSE [%d]: %r", -msg.req, msg)
             if msg.end_err:
@@ -183,7 +213,11 @@ class PacketStream(object):
         logger.debug("WRITE HDR: %s", header)
         logger.debug("WRITE DATA: %s", msg.data)
 
-    def send(self, data, msg_type=PSMessageType.JSON, stream=False, end_err=False, req=None):
+    def send(  # pylint: disable=too-many-arguments
+        self, data, msg_type=PSMessageType.JSON, stream=False, end_err=False, req=None
+    ):
+        """Send data through the packet stream"""
+
         update_counter = False
         if req is None:
             update_counter = True
@@ -205,5 +239,7 @@ class PacketStream(object):
         return handler
 
     def disconnect(self):
+        """Disconnect the stream"""
+
         self._connected = False
         self.connection.disconnect()