wsproto

Author: joente

thingsdb/client/client.py

@@ -380,7 +380,7 @@ async def _ensure_write(
             try:
                 assert self._protocol  # we're connected
                 res = await self._protocol.write(tp, data, is_bin, timeout)
-            except (asyncio.exceptions.CancelledError,
+            except (asyncio.CancelledError,
                     CancelledError, NodeError, AuthError) as e:
                 if timeout and time.time() - start > timeout:
                     msg = str(e) or type(e).__name__

thingsdb/client/wsprotocol.py

@@ -13,6 +13,7 @@
 
 
 class ProtocolWS(BaseProtocol):
+
     """More a wrapper than a true protocol."""
     def __init__(
         self,