fixed extra_headers and ws.open (#145)

* fixed extra_headers and ws.open

* Bump minimum Python requirement to 3.9 for Websockets>=14 and update websocket requirement in pyproject.toml

* Update pytr/api.py

* ruff format

---------

Co-authored-by: Niklas Rosenstein <[email protected]>

Author: nitro424

pyproject.toml

@@ -7,7 +7,7 @@ name = "pytr"
 version = "0.3.1"
 description = "Use TradeRepublic in terminal"
 readme = "README.md"
-requires-python = ">=3.8"
+requires-python = ">=3.9"
 license = { text = "MIT" }
 authors = [
     { name = "marzzzello", email = "[email protected]" }
@@ -30,7 +30,7 @@ dependencies = [
     "pygments",
     "requests_futures",
     "shtab",
-    "websockets>=10.1,<14",
+    "websockets>=14",
     "babel",
 ]
 

pytr/api.py

@@ -263,7 +263,7 @@ def _web_request(self, url_path, payload=None, method="GET"):
         return self._websession.request(method=method, url=f"{self._host}{url_path}", data=payload)
 
     async def _get_ws(self):
-        if self._ws and self._ws.open:
+        if self._ws and self._ws.close_code is None:
             return self._ws
 
         self.log.info("Connecting to websocket ...")
@@ -289,7 +289,9 @@ async def _get_ws(self):
             }
             connect_id = 31
 
-        self._ws = await websockets.connect("wss://api.traderepublic.com", ssl=ssl_context, extra_headers=extra_headers)
+        self._ws = await websockets.connect(
+            "wss://api.traderepublic.com", ssl=ssl_context, additional_headers=extra_headers
+        )
         await self._ws.send(f"connect {connect_id} {json.dumps(connection_message)}")
         response = await self._ws.recv()