Skip to content

Commit 1fa1ebe

Browse files
authored
Fix too-restrictive websocket typing (#276)
1 parent e9f1a6d commit 1fa1ebe

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

simplipy/websocket.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ def connected(self) -> bool:
229229

230230
@staticmethod
231231
def _add_listener(
232-
listener_list: list, callback: Callable[..., None]
232+
listener_list: list, callback: Callable[..., Any]
233233
) -> Callable[..., None]:
234234
"""Add a listener callback to a particular list."""
235235
listener_list.append(callback)
@@ -293,9 +293,7 @@ def _schedule_listener_call(self, listener: Callable[..., Any], *args: Any) -> N
293293
else:
294294
self._loop.call_soon(listener, *args)
295295

296-
def add_connect_listener(
297-
self, callback: Callable[..., None]
298-
) -> Callable[..., None]:
296+
def add_connect_listener(self, callback: Callable[..., Any]) -> Callable[..., None]:
299297
"""Add a listener callback to be called after connecting.
300298
301299
:param callback: The method to call after connecting
@@ -304,7 +302,7 @@ def add_connect_listener(
304302
return self._add_listener(self._connect_listeners, callback)
305303

306304
def add_disconnect_listener(
307-
self, callback: Callable[..., None]
305+
self, callback: Callable[..., Any]
308306
) -> Callable[..., None]:
309307
"""Add a listener callback to be called after disconnecting.
310308
@@ -313,7 +311,7 @@ def add_disconnect_listener(
313311
"""
314312
return self._add_listener(self._disconnect_listeners, callback)
315313

316-
def add_event_listener(self, callback: Callable[..., None]) -> Callable[..., None]:
314+
def add_event_listener(self, callback: Callable[..., Any]) -> Callable[..., None]:
317315
"""Add a listener callback to be called upon receiving an event.
318316
319317
Note that callbacks should expect to receive a WebsocketEvent object as a

0 commit comments

Comments
 (0)