1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
from __future__ import annotations
from twisted.python.failure import Failure
from twisted.web.iweb import IRequest
from twisted.web.resource import Resource
from twisted.web.static import File
from twisted.web.websocket import WebSocketResource, WebSocketTransport
from twisted.internet.task import LoopingCall
class WebSocketDemo:
loop: LoopingCall | None = None
@classmethod
def buildProtocol(cls, request: IRequest) -> WebSocketDemo:
return cls()
def negotiationStarted(self, transport: WebSocketTransport) -> None:
self.transport = transport
self.counter = 0
def negotiationFinished(self) -> None:
def heartbeat() -> None:
self.counter += 1
self.transport.sendTextMessage(f"heartbeat {self.counter}")
self.loop = LoopingCall(heartbeat)
self.loop.start(1.0)
def textMessageReceived(self, data: str) -> None:
self.transport.sendTextMessage(f"reply to {data}")
def connectionLost(self, reason: Failure) -> None:
if self.loop is not None:
self.loop.stop()
# Since WebSocketProtocol is a typing.Protocol and not a class, we must
# provide implementations for all events, even those we don't care about.
def bytesMessageReceived(self, data: bytes) -> None: ...
def pongReceived(self, payload: bytes) -> None: ...
resource = WebSocketResource(WebSocketDemo)
|
