blob: cdc45db5b94748e515581eeb4a701c224ea4950b (
plain) (
blame)
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
|
# coding=utf-8
import logging
from docutils.core import publish_parts
from geventwebsocket.websocket import WebSocket
from pysite.websockets import WS
class RSTWebsocket(WS):
path = "/rst"
name = "ws.rst"
def __init__(self, socket: WebSocket):
super().__init__(socket)
self.log = logging.getLogger()
def on_open(self):
self.log.debug("RST | WS opened.")
self.send("Hey, welcome!")
def on_message(self, message):
self.log.debug(f"RST | Message: {message}")
try:
data = publish_parts(
source=message, writer_name="html5", settings_overrides={"traceback": True, "halt_level": 2}
)["html_body"]
except Exception as e:
self.log.exception("Parsing error")
data = str(e)
self.send(data)
def on_close(self):
self.log.debug("RST | WS closed.")
|
