Bump websockets version

.github/workflows/pythonpublish.yml

@@ -20,7 +20,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip setuptools wheel twine
-          python -m pip install aiohttp backoff coverage coveralls pydantic websockets==8.1 yarl
+          python -m pip install aiohttp backoff coverage coveralls pydantic websockets>=9.1 yarl
       - name: Build and publish
         env:
           TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}

.github/workflows/pythonunittest.yml

@@ -20,7 +20,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install aiohttp backoff coverage coveralls pydantic websockets==8.1 yarl
+          pip install aiohttp backoff coverage coveralls pydantic websockets>=9.1 yarl
       - name: Run tests
         env:
           SERVICE_ID: ${{secrets.REPO_SERVICE_ID}}

auraxium/event/_client.py

@@ -4,15 +4,16 @@
 import logging
 from typing import (Any, Callable, Coroutine, Dict, Iterator, List, Optional, Type, TypeVar, Union,
                     cast, overload)
+
 import backoff
 import pydantic
-
 import websockets
+import websockets.exceptions
+from websockets.legacy import client as ws_client
 
 from .._client import Client
 from ..models import Event
 from ..types import CensusData
-
 from ._trigger import Trigger
 
 __all__ = [
@@ -55,7 +56,7 @@ class EventClient(Client):
        registered for the client.
 
     .. attribute:: websocket
-       :type: websockets.client.WebSocketClientProtocol | None
+       :type: websockets.client.legacy.WebSocketClientProtocol | None
 
        The websocket client used for the real-time event stream. This
        will be automatically opened and closed by the client as event
@@ -65,7 +66,7 @@ class EventClient(Client):
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         super().__init__(*args, **kwargs)
         self.triggers: List[Trigger] = []
-        self.websocket: Optional[websockets.WebSocketClientProtocol] = None
+        self.websocket: Optional[ws_client.WebSocketClientProtocol] = None
         self._connect_lock = asyncio.Lock()
         self._connected: bool = False
         self._endpoint_status: Dict[str, bool] = {}
@@ -263,7 +264,7 @@ async def _connection_handler(self) -> None:
         """
         _log.info('Connecting to WebSocket endpoint...')
         url = f'{_ESS_ENDPOINT}?environment=ps2&service-id={self.service_id}'
-        async with websockets.connect(url) as websocket:
+        async with ws_client.connect(url) as websocket:
             self.websocket = websocket
             _log.info('Connected to %s?environment=ps2&service-id=XXX',
                       _ESS_ENDPOINT)

docs/requirements.txt

@@ -4,4 +4,4 @@ backoff
 pydantic
 requests
 yarl
-websockets == 8.1
+websockets >= 9.1

setup.py

@@ -26,7 +26,7 @@
                                    'backoff',
                                    'pydantic',
                                    'yarl',
-                                   'websockets==8.1'],
+                                   'websockets>=9.1'],
                  license='MIT',
                  include_package_data=True,
                  zip_safe=False)