From 00ba86847037838ae01d4b509c8a3a60cf6f3d8c Mon Sep 17 00:00:00 2001
From: Gareth Coles <gdude2002@gmail.com>
Date: Thu, 5 Apr 2018 10:02:03 +0100
Subject: RST test error catching

---
 pysite/views/ws/rst.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

(limited to 'pysite/views/ws/rst.py')

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.")
-- 
cgit v1.2.3