ws improvements

Author: joente

thingsdb/client/wsprotocol.py

@@ -33,8 +33,8 @@ async def connect(self, uri, ssl: SSLContext | None):
 
     async def _recv_loop(self):
         try:
-            while True:
-                data: bytes = await self._proto.recv()  # type: ignore
+            while self._proto:
+                data: bytes = await self._proto.recv(decode=False)
                 pkg = None
                 try:
                     pkg = Package(data)
@@ -50,34 +50,35 @@ async def _recv_loop(self):
 
         except ConnectionClosed as exc:
             self.cancel_requests()
-            self._proto = None  # type: ignore
+            self._proto = None
             self._on_connection_lost(self, exc)  # type: ignore
 
     def _write(self, data: Any):
         if self._proto is None:
             raise ConnectionError('no connection')
-        asyncio.create_task(self._proto.send(data))  # type: ignore
+        asyncio.create_task(self._proto.send(data))
 
     def close(self):
         self._is_closing = True
         if self._proto:
-            asyncio.create_task(self._proto.close())  # type: ignore
+            asyncio.create_task(self._proto.close())
 
     def is_closing(self) -> bool:
         return self._is_closing
 
     async def wait_closed(self):
         if self._proto:
-            await self._proto.wait_closed()  # type: ignore
+            await self._proto.wait_closed()
 
     async def close_and_wait(self):
         if self._proto:
-            await self._proto.close()  # type: ignore
+            await self._proto.close()
 
     def info(self) -> Any:
-        return self._proto.transport.get_extra_info(  # type: ignore
-            'socket',
-            None)
+        if self._proto:
+            return self._proto.transport.get_extra_info(
+                'socket',
+                None)
 
     def is_connected(self) -> bool:
         return self._proto is not None

thingsdb/version.py

@@ -1 +1 @@
-__version__ = '1.3.0'
+__version__ = '1.3.1'