RST test error catching

1 files changed, 8 insertions, 3 deletions

diff --git a/pysite/views/ws/rst.py b/pysite/views/ws/rst.py
index 52bdf09b..49afaa88 100644
--- a/pysite/views/ws/rst.py
+++ b/pysite/views/ws/rst.py
@@ -21,11 +21,16 @@ class RSTWebsocket(WS):
 
     def on_message(self, message):
         self.log.debug(f"RST | Message: {message}")
-        self.send(
-            publish_parts(
+
+        try:
+            data = publish_parts(
                 source=message, writer_name="html5", settings_overrides={"traceback": True}
             )["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.")