From 194a0abbac234dd22b45e8351e7762219aea64f4 Mon Sep 17 00:00:00 2001
From: Christopher Banck <vchrisb@users.noreply.github.com>
Date: Mon, 4 Dec 2023 09:22:23 +0100
Subject: [PATCH] update typing for WebSocketApp

---
 e3dc/_e3dc_rscp_web.py | 28 +++++++++-------------------
 1 file changed, 9 insertions(+), 19 deletions(-)

diff --git a/e3dc/_e3dc_rscp_web.py b/e3dc/_e3dc_rscp_web.py
index 67e5c06..5f0ab5f 100644
--- a/e3dc/_e3dc_rscp_web.py
+++ b/e3dc/_e3dc_rscp_web.py
@@ -13,7 +13,7 @@
 from typing import Any, Callable, Tuple
 
 import tzlocal
-from websocket import ABNF, WebSocketApp  # pyright: ignore [reportMissingTypeStubs]
+from websocket import ABNF, WebSocketApp  # pyright: ignore [reportPrivateImportUsage]
 
 from ._rscpLib import (
     rscpDecode,
@@ -120,11 +120,9 @@ def __init__(
         self.serialNumberWithPrefix = serialNumberWithPrefix.encode("utf-8")
         self.ws = WebSocketApp(
             REMOTE_ADDRESS,
-            on_message=lambda _, msg: self.on_message(  # pyright: ignore [reportUnknownLambdaType]
-                msg  # pyright: ignore [reportUnknownArgumentType]
-            ),
-            on_close=lambda _: self.reset(),  # pyright: ignore [reportUnknownLambdaType]
-            on_error=lambda _: self.reset(),  # pyright: ignore [reportUnknownLambdaType]
+            on_message=lambda _, msg: self.on_message(msg),
+            on_close=lambda _ws, _, __: self.reset(),
+            on_error=lambda _ws, _: self.reset(),
         )
         self.reset()
 
@@ -163,9 +161,7 @@ def buildVirtualConn(self):
         )
 
         # print("--------------------- Sending virtual conn")
-        self.ws.send(
-            virtualConn, ABNF.OPCODE_BINARY  # pyright: ignore [reportGeneralTypeIssues]
-        )
+        self.ws.send(virtualConn, ABNF.OPCODE_BINARY)
 
     def respondToINFORequest(
         self, decoded: Tuple[str | int | RscpTag, str | int | RscpType, Any]
@@ -255,9 +251,7 @@ def registerConnectionHandler(
                 ],
             )
         )
-        self.ws.send(
-            reply, ABNF.OPCODE_BINARY  # pyright: ignore [reportGeneralTypeIssues]
-        )
+        self.ws.send(reply, ABNF.OPCODE_BINARY)
 
     def on_message(self, message: bytes):
         """Method to handle a received message."""
@@ -278,7 +272,7 @@ def on_message(self, message: bytes):
                 rscpEncode(RscpTag.SERVER_PING, RscpType.NoneType, None)
             )
             self.ws.send(
-                pingFrame,  # pyright: ignore [reportGeneralTypeIssues]
+                pingFrame,
                 ABNF.OPCODE_BINARY,
             )
             return
@@ -328,9 +322,7 @@ def on_message(self, message: bytes):
             )
 
             self.ws.send(
-                rscpFrame(
-                    responseContainer
-                ),  # pyright: ignore [reportGeneralTypeIssues]
+                rscpFrame(responseContainer),
                 ABNF.OPCODE_BINARY,
             )
 
@@ -406,9 +398,7 @@ def _sendRequest_internal(
             )
         )
 
-        self.ws.send(
-            outerFrame, ABNF.OPCODE_BINARY  # pyright: ignore [reportGeneralTypeIssues]
-        )
+        self.ws.send(outerFrame, ABNF.OPCODE_BINARY)
 
     def connect(self):
         """Connect to E3DC system."""