Update handlers.py

If message is bytes, we should set binary=True
jupyter · May 8, 2019 · 46b3cd2 · 46b3cd2
1 parent 2d3095a
commit 46b3cd2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nb2kg/handlers.py b/nb2kg/handlers.py
@@ -115,6 +115,8 @@ def on_message(self, message):
     def write_message(self, message, binary=False):
         """Send message back to notebook client.  This is called via callback from self.gateway._read_messages."""
         if self.ws_connection:  # prevent WebSocketClosedError
+            if isinstance(message, bytes):
+                binary = True
             super(WebSocketChannelsHandler, self).write_message(message, binary=binary)
         elif self.log.isEnabledFor(logging.DEBUG):
             msg_summary = WebSocketChannelsHandler._get_message_summary(json_decode(utf8(message)))