fix: adapt to new websockets API (#234)

* wip

* pycodestyle

* update dependencies

* skl2onnx

* use ruff

* apply formatter

* apply lint auto fixes

* manually apply lints

* change check

* ruff ci from branch

* fix websockets API

* revert pyproject

* fix doc comment

* set ci branch to main again

Author: ChristianBeilschmidt

.github/workflows/ci.yml

@@ -12,7 +12,7 @@ on:
 
 jobs:
   check:
-    uses: geo-engine/geoengine-python/.github/workflows/test-python.yml@ruff
+    uses: geo-engine/geoengine-python/.github/workflows/test-python.yml@main
 
     strategy:
       fail-fast: false
@@ -28,7 +28,7 @@ jobs:
   # Checks the library using minimum version resolution
   # `uv` has this feature built-in, c.f. https://github.com/astral-sh/uv
   check-min-version:
-    uses: geo-engine/geoengine-python/.github/workflows/test-python.yml@ruff
+    uses: geo-engine/geoengine-python/.github/workflows/test-python.yml@main
 
     with:
       python-version: "3.10"

geoengine/workflow.py

@@ -25,6 +25,7 @@
 import requests as req
 import rioxarray
 import websockets
+import websockets.asyncio.client
 import xarray as xr
 from owslib.util import Authentication, ResponseWrapper
 from owslib.wcs import WebCoverageService
@@ -622,7 +623,7 @@ async def raster_stream(
 
         async with websockets.asyncio.client.connect(
             uri=self.__replace_http_with_ws(url),
-            extra_headers=session.auth_header,
+            additional_headers=session.auth_header,
             open_timeout=open_timeout,
             max_size=None,
         ) as websocket:

pyproject.toml

@@ -29,7 +29,7 @@ dependencies = [
     "rioxarray >=0.9.1, <0.20",
     "StrEnum >=0.4.6,<0.5",               # TODO: use from stdlib when `python_requires = >=3.11`
     "vega >= 3.5,<4.2",
-    "websockets >= 14.0,<16",
+    "websockets >= 14.2,<16",
     "xarray >=0.19,<2025.8",
     "urllib3 >= 2.1, < 2.6",
     "pydantic >= 2.10.6, < 2.12",