Remove json_clean

Author: martinRenou

ipykernel/comm/comm.py

@@ -8,7 +8,6 @@
 from traitlets.config import LoggingConfigurable
 from ipykernel.kernelbase import Kernel
 
-from ipykernel.jsonutil import json_clean
 from traitlets import Instance, Unicode, Bytes, Bool, Dict, Any, default
 
 
@@ -62,10 +61,10 @@ def _publish_msg(self, msg_type, data=None, metadata=None, buffers=None, **keys)
         """Helper for sending a comm message on IOPub"""
         data = {} if data is None else data
         metadata = {} if metadata is None else metadata
-        content = json_clean(dict(data=data, comm_id=self.comm_id, **keys))
+        content = dict(data=data, comm_id=self.comm_id, **keys)
         self.kernel.session.send(self.kernel.iopub_socket, msg_type,
             content,
-            metadata=json_clean(metadata),
+            metadata=metadata,
             parent=self.kernel.get_parent("shell"),
             ident=self.topic,
             buffers=buffers,

ipykernel/datapub.py

@@ -12,7 +12,6 @@
 
 from traitlets.config import Configurable
 from traitlets import Instance, Dict, CBytes, Any
-from ipykernel.jsonutil import json_clean
 try:
     # available since ipyparallel 5.0.0
     from ipyparallel.serialize import serialize_object
@@ -46,7 +45,7 @@ def publish_data(self, data):
             buffer_threshold=session.buffer_threshold,
             item_threshold=session.item_threshold,
         )
-        content = json_clean(dict(keys=list(data.keys())))
+        content = dict(keys=list(data.keys()))
         session.send(self.pub_socket, 'data_message', content=content,
             parent=self.parent_header,
             buffers=buffers,
@@ -66,6 +65,6 @@ def publish_data(data):
         DeprecationWarning,
         stacklevel=2
     )
-    
+
     from ipykernel.zmqshell import ZMQInteractiveShell
     ZMQInteractiveShell.instance().data_pub.publish_data(data)

ipykernel/displayhook.py

@@ -7,7 +7,6 @@
 import sys
 
 from IPython.core.displayhook import DisplayHook
-from ipykernel.jsonutil import encode_images, json_clean
 from traitlets import Instance, Dict, Any
 from jupyter_client.session import extract_header, Session
 
@@ -68,7 +67,7 @@ def write_output_prompt(self):
         self.msg['content']['execution_count'] = self.prompt_count
 
     def write_format_data(self, format_dict, md_dict=None):
-        self.msg['content']['data'] = json_clean(encode_images(format_dict))
+        self.msg['content']['data'] = format_dict
         self.msg['content']['metadata'] = md_dict
 
     def finish_displayhook(self):

ipykernel/inprocess/ipkernel.py

@@ -8,7 +8,6 @@
 import sys
 
 from IPython.core.interactiveshell import InteractiveShellABC
-from ipykernel.jsonutil import json_clean
 from traitlets import Any, Enum, Instance, List, Type, default
 from ipykernel.ipkernel import IPythonKernel
 from ipykernel.zmqshell import ZMQInteractiveShell
@@ -98,7 +97,7 @@ def _input_request(self, prompt, ident, parent, password=False):
         sys.stdout.flush()
 
         # Send the input request.
-        content = json_clean(dict(prompt=prompt, password=password))
+        content = dict(prompt=prompt, password=password)
         msg = self.session.msg('input_request', content, parent)
         for frontend in self.frontends:
             if frontend.session.session == parent['header']['session']:

ipykernel/jsonutil.py

@@ -70,7 +70,9 @@ def encode_images(format_dict):
 
 
 def json_clean(obj):
-    """Clean an object to ensure it's safe to encode in JSON.
+    """Deprecated, this is a no-op now.
+
+    Clean an object to ensure it's safe to encode in JSON.
 
     Atomic, immutable objects are returned unmodified.  Sets and tuples are
     converted to lists, lists are copied and dicts are also copied.
@@ -91,46 +93,4 @@ def json_clean(obj):
         it simply sanitizes it so that there will be no encoding errors later.
 
     """
-    # types that are 'atomic' and ok in json as-is.
-    atomic_ok = (bool, str, type(None))
-
-    # containers that we need to convert into lists
-    container_to_list = (list, tuple, set, types.GeneratorType)
-
-    # Since bools are a subtype of Integrals, which are a subtype of Reals,
-    # we have to check them in that order.
-
-    if isinstance(obj, atomic_ok):
-        return obj
-
-    if isinstance(obj, numbers.Integral):
-        # cast int to int, in case subclasses override __str__ (e.g. boost enum, #4598)
-        return int(obj)
-
-    if isinstance(obj, numbers.Real):
-        # cast out-of-range floats to their reprs
-        if math.isnan(obj) or math.isinf(obj):
-            return repr(obj)
-        return float(obj)
-
-    if isinstance(obj, bytes):
-        # unanmbiguous binary data is base64-encoded
-        # (this probably should have happened upstream)
-        return b2a_base64(obj).decode('ascii')
-
-    if isinstance(obj, container_to_list) or (
-        hasattr(obj, '__iter__') and hasattr(obj, next_attr_name)):
-        return [json_clean(x) for x in obj]
-
-    if isinstance(obj, dict):
-        # If all OK, proceed by making the new dict that will be json-safe
-        out = {}
-        for k, v in obj.items():
-            out[str(k)] = json_clean(v)
-        return out
-
-    if isinstance(obj, datetime):
-        return obj.strftime(ISO8601)
-
-    # we don't understand it, it's probably an unserializable object
-    raise ValueError("Can't clean for JSON: %r" % obj)
+    return obj

ipykernel/kernelbase.py

@@ -30,7 +30,6 @@
 
 from traitlets.config.configurable import SingletonConfigurable
 from IPython.core.error import StdinNotImplementedError
-from ipykernel.jsonutil import json_clean
 from traitlets import (
     Any, Instance, Float, Dict, List, Set, Integer, Unicode, Bool,
     observe, default
@@ -656,7 +655,6 @@ async def execute_request(self, stream, ident, parent):
             time.sleep(self._execute_sleep)
 
         # Send the reply.
-        reply_content = json_clean(reply_content)
         metadata = self.finish_metadata(parent, metadata, reply_content)
 
         reply_msg = self.session.send(stream, 'execute_reply',
@@ -683,7 +681,6 @@ async def complete_request(self, stream, ident, parent):
         if inspect.isawaitable(matches):
             matches = await matches
 
-        matches = json_clean(matches)
         self.session.send(stream, "complete_reply", matches, parent, ident)
 
     def do_complete(self, code, cursor_pos):
@@ -706,7 +703,6 @@ async def inspect_request(self, stream, ident, parent):
             reply_content = await reply_content
 
         # Before we send this object over, we scrub it for JSON usage
-        reply_content = json_clean(reply_content)
         msg = self.session.send(stream, 'inspect_reply',
                                 reply_content, parent, ident)
         self.log.debug("%s", msg)
@@ -723,7 +719,6 @@ async def history_request(self, stream, ident, parent):
         if inspect.isawaitable(reply_content):
             reply_content = await reply_content
 
-        reply_content = json_clean(reply_content)
         msg = self.session.send(stream, 'history_reply',
                                 reply_content, parent, ident)
         self.log.debug("%s", msg)
@@ -813,7 +808,6 @@ async def is_complete_request(self, stream, ident, parent):
         reply_content = self.do_is_complete(code)
         if inspect.isawaitable(reply_content):
             reply_content = await reply_content
-        reply_content = json_clean(reply_content)
         reply_msg = self.session.send(stream, 'is_complete_reply',
                                       reply_content, parent, ident)
         self.log.debug("%s", reply_msg)
@@ -829,7 +823,6 @@ async def debug_request(self, stream, ident, parent):
         reply_content = self.do_debug_request(content)
         if inspect.isawaitable(reply_content):
             reply_content = await reply_content
-        reply_content = json_clean(reply_content)
         reply_msg = self.session.send(stream, 'debug_reply', reply_content,
                                       parent, ident)
         self.log.debug("%s", reply_msg)
@@ -1001,7 +994,7 @@ def _input_request(self, prompt, ident, parent, password=False):
                     raise
 
         # Send the input request.
-        content = json_clean(dict(prompt=prompt, password=password))
+        content = dict(prompt=prompt, password=password)
         self.session.send(self.stdin_socket, 'input_request', content, parent,
                           ident=ident)
 

ipykernel/tests/test_jsonutil.py

@@ -35,7 +35,6 @@
     get_connection_file, get_connection_info, connect_qtconsole
 )
 from IPython.utils import openpy
-from ipykernel.jsonutil import json_clean, encode_images
 from IPython.utils.process import arg_split, system
 from traitlets import (
     Instance, Type, Dict, CBool, CBytes, Any, default, observe
@@ -113,7 +112,7 @@ def publish(
             transient = {}
         self._validate_data(data, metadata)
         content = {}
-        content['data'] = encode_images(data)
+        content['data'] = data
         content['metadata'] = metadata
         content['transient'] = transient
 
@@ -123,7 +122,7 @@ def publish(
         # in order to put it through the transform
         # hooks before potentially sending.
         msg = self.session.msg(
-            msg_type, json_clean(content),
+            msg_type, content,
             parent=self.parent_header
         )
 
@@ -552,7 +551,7 @@ def _showtraceback(self, etype, evalue, stb):
         dh.session.send(
             dh.pub_socket,
             "error",
-            json_clean(exc_content),
+            exc_content,
             dh.parent_header,
             ident=topic,
         )