Optimize processing of websocket messages (#57)

Switch to asyncio.timeout/async_timeout to avoid task creation
- asyncio.wait_for creates a task which is not needed here
- the wait_for implementation will eventually be replaced but not until 3.12 in python/cpython#98518

Use orjson for parsing json

Author: bdraco

sense_energy/asyncsenseable.py

@@ -1,13 +1,19 @@
 import asyncio
 import json
 import ssl
+import sys
 
 import aiohttp
+import orjson
 import websockets
 
 from .sense_api import *
 from .sense_exceptions import *
 
+if sys.version_info[:2] < (3, 11):
+    from async_timeout import timeout as asyncio_timeout
+else:
+    from asyncio import timeout as asyncio_timeout
 
 class ASyncSenseable(SenseableBase):
     def __init__(
@@ -145,11 +151,12 @@ async def async_realtime_stream(self, callback=None, single=False):
         async with websockets.connect(url, ssl=self.ssl_context) as ws:
             while True:
                 try:
-                    message = await asyncio.wait_for(ws.recv(), timeout=self.wss_timeout)
+                    async with asyncio_timeout(self.wss_timeout):
+                        message = await ws.recv()
                 except asyncio.TimeoutError as ex:
                     raise SenseAPITimeoutException("API websocket timed out") from ex
 
-                result = json.loads(message)
+                result = orjson.loads(message)
                 if result.get("type") == "realtime_update":
                     data = result["payload"]
                     self._set_realtime(data)

setup.py

@@ -8,6 +8,8 @@
     name = 'sense_energy',
     packages = ['sense_energy'], 
     install_requires=[
+        'async_timeout>=3.0.1',
+        'orjson',
         'requests',
         'websocket-client',
         'websockets;python_version>="3.5"',