Refactor to use plotly.py to_json method for json encoding

This will provide future compatibility with the plotly.py 5.0 JSON engine, and accelerated orjson encoder.

Author: jonmmease

repos/kaleido/py/kaleido/scopes/base.py

@@ -14,9 +14,6 @@
 
 
 class BaseScope(object):
-    # Subclasses may override to specify a custom JSON encoder for input data
-    _json_encoder = None
-
     # Tuple of class properties that will be passed as command-line
     # flags to configure scope
     _scope_flags = ()
@@ -274,6 +271,9 @@ def chromium_args(self, val):
         self._chromium_args = tuple(val)
         self._shutdown_kaleido()
 
+    def _json_dumps(self, val):
+        return json.dumps(val)
+
     def _perform_transform(self, data, **kwargs):
         """
         Transform input data using the current scope, returning dict response with error code
@@ -287,9 +287,7 @@ def _perform_transform(self, data, **kwargs):
         self._ensure_kaleido()
 
         # Perform export
-        export_spec = json.dumps(
-            dict(kwargs, data=data),
-            cls=self._json_encoder).encode('utf-8')
+        export_spec = self._json_dumps(dict(kwargs, data=data)).encode('utf-8')
 
         # Write to process and read result within a lock so that can be
         # sure we're reading the response to our request

repos/kaleido/py/kaleido/scopes/plotly.py

@@ -1,6 +1,6 @@
 from __future__ import absolute_import
 from kaleido.scopes.base import BaseScope, which
-from _plotly_utils.utils import PlotlyJSONEncoder
+import plotly.io as pio
 import base64
 import os
 from pathlib import Path
@@ -10,7 +10,6 @@ class PlotlyScope(BaseScope):
     """
     Scope for transforming Plotly figures to static images
     """
-    _json_encoder = PlotlyJSONEncoder
     _all_formats = ("png", "jpg", "jpeg", "webp", "svg", "pdf", "eps", "json")
     _text_formats = ("svg", "json", "eps")
 
@@ -73,6 +72,9 @@ def _initialize_mathax(self, mathjax=None):
     def scope_name(self):
         return "plotly"
 
+    def _json_dumps(self, val):
+        return pio.to_json(val, validate=False, remove_uids=False)
+
     def transform(self, figure, format=None, width=None, height=None, scale=None):
         """
         Convert a Plotly figure into a static image