safely import simplejson for python 2.6

Author: theengineear

plotly/exceptions.py

@@ -15,9 +15,14 @@
 
 
 """
-import json
+import sys
 import six
 
+if sys.version[:3] == '2.6':
+    import simplejson as json
+else:
+    import json
+
 ## Base Plotly Error ##
 
 class PlotlyError(Exception):

plotly/grid_objs/grid_objs.py

@@ -5,12 +5,16 @@
 """
 from __future__ import absolute_import
 
-
-import json
+import sys
 from collections import MutableSequence
 from plotly import exceptions
 from plotly import utils
 
+if sys.version[:3] == '2.6':
+    import simplejson as json
+else:
+    import json
+
 __all__ = None
 
 

plotly/plotly/plotly.py

@@ -17,7 +17,6 @@
 from __future__ import absolute_import
 
 import sys
-import json
 import warnings
 import copy
 import os
@@ -30,6 +29,11 @@
 else:
     from urllib.parse import urlparse
 
+if sys.version[:3] == '2.6':
+    import simplejson as json
+else:
+    import json
+
 from plotly.plotly import chunked_requests
 from plotly import utils
 from plotly import tools

plotly/tests/test_core/test_get_figure/test_get_figure.py

@@ -10,7 +10,6 @@
 from plotly import exceptions
 from nose.tools import raises
 import six
-import json
 
 from unittest import TestCase
 

plotly/tests/test_core/test_get_requests/test_get_requests.py

@@ -8,8 +8,13 @@
 
 import requests
 import copy
-import json
 import six
+import sys
+
+if sys.version[:3] == '2.6':
+    import simplejson as json
+else:
+    import json
 
 default_headers = {'plotly-username': '',
                    'plotly-apikey': '',

plotly/tests/test_core/test_utils/test_utils.py

@@ -1,8 +1,13 @@
-import json
+import sys
 from unittest import TestCase
 
 from plotly.utils import PlotlyJSONEncoder
 
+if sys.version[:3] == '2.6':
+    import simplejson as json
+else:
+    import json
+
 
 class TestJSONEncoder(TestCase):
 

plotly/widgets/graph_widget.py

@@ -1,6 +1,6 @@
 from collections import deque
-import json
 import uuid
+import sys
 
 # TODO: protected imports?
 from IPython.html import widgets
@@ -11,6 +11,12 @@
 from plotly.graph_objs import Figure
 from pkg_resources import resource_string
 
+# even though python 2.6 wouldn't be able to run *any* of this...
+if sys.version[:3] == '2.6':
+    import simplejson as json
+else:
+    import json
+
 # Load JS widget code
 # No officially recommended way to do this in any other way
 # http://mail.scipy.org/pipermail/ipython-dev/2014-April/013835.html