diff --git a/stubs/cyclone/websocket.pyi b/stubs/cyclone/websocket.pyi --- a/stubs/cyclone/websocket.pyi +++ b/stubs/cyclone/websocket.pyi @@ -4,7 +4,7 @@ import cyclone.escape import cyclone.web -from typing import Any +from typing import Any, Dict, Union class _NotEnoughFrame(Exception): ... @@ -17,8 +17,8 @@ class WebSocketHandler(cyclone.web.Reque def headersReceived(self) -> None: ... def connectionMade(self, *args: Any, **kwargs: Any) -> None: ... def connectionLost(self, reason: Any) -> None: ... - def messageReceived(self, message: Any) -> None: ... - def sendMessage(self, message: Any) -> None: ... + def messageReceived(self, message: bytes) -> None: ... + def sendMessage(self, message: Union[str, Dict]) -> None: ... def forbidConnection(self, message: Any): ... class WebSocketProtocol: @@ -28,20 +28,20 @@ class WebSocketProtocol: def __init__(self, handler: Any) -> None: ... def acceptConnection(self) -> None: ... def rawDataReceived(self, data: Any) -> None: ... - def sendMessage(self, message: Any) -> None: ... + def sendMessage(self, message: Union[str, Dict]) -> None: ... class WebSocketProtocol17(WebSocketProtocol): def __init__(self, handler: Any) -> None: ... def acceptConnection(self) -> None: ... def rawDataReceived(self, data: Any) -> None: ... - def sendMessage(self, message: Any, code: int = ...) -> None: ... + def sendMessage(self, message: Union[str, Dict], code: int = ...) -> None: ... class WebSocketProtocol76(WebSocketProtocol): def __init__(self, handler: Any) -> None: ... def acceptConnection(self) -> None: ... def rawDataReceived(self, data: Any) -> None: ... def close(self) -> None: ... - def sendMessage(self, message: Any) -> None: ... + def sendMessage(self, message: Union[str, Dict]) -> None: ... class FrameDecodeError(Exception): ...