WIP delaying import in io module.

Importing io by itself is now < 50ms
plotly · jonmmease · Apr 12, 2019 · Mar 16, 2019 · Mar 16, 2019 · Mar 16, 2019
commit 049e304e8725d3dbddecd77a85d4bb66e192c25a
diff --git a/plotly/__init__.py b/plotly/__init__.py
@@ -28,15 +28,25 @@
 
 from __future__ import absolute_import
 
-from plotly import (plotly, dashboard_objs, graph_objs, grid_objs, tools,
-                    utils, session, offline, colors, io)
-from plotly.version import __version__
+from plotly import (
+    # plotly,
+    # dashboard_objs,
+    # graph_objs,
+    # grid_objs,
+    # tools,
+    # utils,
+    # session,
+    # offline,
+    # colors,
+    io
+)
+# from plotly.version import __version__
 from _plotly_future_ import _future_flags
-
-from ._version import get_versions
-__version__ = get_versions()['version']
-del get_versions
-
-# Set default template here to make sure import process is complet
-if 'template_defaults' in _future_flags:
-    io.templates._default = 'plotly'
+#
+# from ._version import get_versions
+# __version__ = get_versions()['version']
+# del get_versions
+
+# # Set default template here to make sure import process is complete
+# if 'template_defaults' in _future_flags:
+#     io.templates._default = 'plotly'
diff --git a/plotly/io/_base_renderers.py b/plotly/io/_base_renderers.py
@@ -6,9 +6,9 @@
 
 import six
 from plotly.io import to_json, to_image
-from plotly import utils, optional_imports
+from plotly import optional_imports
+
 from plotly.io._orca import ensure_server
-from plotly.offline.offline import _get_jconfig, get_plotlyjs
 
 ipython_display = optional_imports.get_module('IPython.display')
 IPython = optional_imports.get_module('IPython')
@@ -80,6 +80,7 @@ def __init__(self, config=None):
         self.config = dict(config) if config else {}
 
     def to_mimebundle(self, fig_dict):
+        from plotly.offline.offline import _get_jconfig
         config = _get_jconfig(self.config)
         if config:
             fig_dict['config'] = config
@@ -267,6 +268,7 @@ def activate(self):
                    mathjax_config=_mathjax_config)
 
             else:
+                from plotly.offline.offline import get_plotlyjs
                 # If not connected then we embed a copy of the plotly.js
                 # library in the notebook
                 script = """\

diff --git a/plotly/io/_html.py b/plotly/io/_html.py
@@ -6,9 +6,6 @@
 import six
 
 from plotly.io._utils import validate_coerce_fig_to_dict
-from plotly.offline.offline import _get_jconfig, get_plotlyjs
-from plotly import utils
-
 
 # Build script to set global PlotlyConfig object. This must execute before
 # plotly.js is loaded.
@@ -118,6 +115,7 @@ def to_html(fig,
     plotdivid = str(uuid.uuid4())
 
     # ## Serialize figure ##
+    from plotly import utils
     jdata = json.dumps(
         fig_dict.get('data', []), cls=utils.PlotlyJSONEncoder)
     jlayout = json.dumps(
@@ -130,6 +128,7 @@ def to_html(fig,
         jframes = None
 
     # ## Serialize figure config ##
+    from plotly.offline.offline import _get_jconfig
     config = _get_jconfig(config)
 
     # Check whether we should add responsive
@@ -233,6 +232,7 @@ def to_html(fig,
                url=include_plotlyjs_orig)
 
     elif include_plotlyjs:
+        from plotly.offline.offline import get_plotlyjs
         load_plotlyjs = """\
         {win_config}
         <script type="text/javascript">{plotlyjs}</script>\
@@ -438,6 +438,7 @@ def write_html(fig,
             os.path.dirname(file), 'plotly.min.js')
 
         if not os.path.exists(bundle_path):
+            from plotly.offline.offline import get_plotlyjs
             with open(bundle_path, 'w') as f:
                 f.write(get_plotlyjs())
 

diff --git a/plotly/io/_json.py b/plotly/io/_json.py
@@ -3,11 +3,10 @@
 from six import string_types
 import json
 
-from plotly.utils import PlotlyJSONEncoder
+
 from plotly.io._utils import (validate_coerce_fig_to_dict,
                               validate_coerce_output_type)
 
-
 def to_json(fig,
             validate=True,
             pretty=False,
@@ -36,6 +35,8 @@ def to_json(fig,
     str
         Representation of figure as a JSON string
     """
+    from plotly.utils import PlotlyJSONEncoder
+
     # Validate figure
     # ---------------
     fig_dict = validate_coerce_fig_to_dict(fig, validate)

diff --git a/plotly/io/_orca.py b/plotly/io/_orca.py
@@ -11,7 +11,6 @@
 from copy import copy
 from contextlib import contextmanager
 
-import requests
 import retrying
 from six import string_types
 
@@ -707,7 +706,7 @@ def __repr__(self):
 
 constants
 ---------
-    plotlyjs: {plotlyjs} 
+    plotlyjs: {plotlyjs}
     config_file: {config_file}
 
 """.format(port=self.port,
@@ -822,7 +821,7 @@ def __repr__(self):
     port: {port}
     pid: {pid}
     command: {command}
-    
+
 """.format(executable=self.executable,
            version=self.version,
            port=self.port,
@@ -910,7 +909,7 @@ def validate_executable():
 Alternatively, see other installation methods in the orca project README at
 https://github.com/plotly/orca.
 
-After installation is complete, no further configuration should be needed. 
+After installation is complete, no further configuration should be needed.
 
 If you have installed orca, then for some reason plotly.py was unable to
 locate it. In this case, set the `plotly.io.orca.config.executable`
@@ -922,7 +921,7 @@ def validate_executable():
 If it is successful then you may want to save this configuration so that it
 will be applied automatically in future sessions. You can do this as follows:
 
-    >>> plotly.io.orca.config.save() 
+    >>> plotly.io.orca.config.save()
 
 If you're still having trouble, feel free to ask for help on the forums at
 https://community.plot.ly/c/api/python
@@ -1038,7 +1037,7 @@ def validate_executable():
 The error encountered is that no version was reported by the orca executable.
 Here is the command that plotly.py ran to request the version:
 
-    $ {executable} --version  
+    $ {executable} --version
 """.format(executable=executable))
     else:
         version_result = version_result.decode()
@@ -1153,7 +1152,7 @@ def ensure_server():
 
 Install using pip:
     $ pip install psutil
-    
+
 Install using conda:
     $ conda install psutil
 """)
@@ -1224,12 +1223,13 @@ def request_image_with_retrying(**kwargs):
     Helper method to perform an image request to a running orca server process
     with retrying logic.
     """
+    from requests import post
     server_url = 'http://{hostname}:{port}'.format(
         hostname='localhost', port=orca_state['port'])
 
     request_params = {k: v for k, v, in kwargs.items() if v is not None}
     json_str = json.dumps(request_params, cls=plotly.utils.PlotlyJSONEncoder)
-    response = requests.post(server_url + '/', data=json_str)
+    response = post(server_url + '/', data=json_str)
     return response
 
 
@@ -1351,7 +1351,7 @@ def to_image(fig,
 
 {info}
 plotly.py will attempt to start the local server process again the next time
-an image export operation is performed. 
+an image export operation is performed.
 """.format(info=status_str))
 
     # Check response
@@ -1393,7 +1393,7 @@ def to_image(fig,
 
 >>> plotly.io.orca.config.mapbox_access_token = 'pk.abc...'
 
-If you would like this token to be applied automatically in 
+If you would like this token to be applied automatically in
 future sessions, then save your orca configuration as follows:
 
 >>> plotly.io.orca.config.save()
@@ -1402,9 +1402,9 @@ def to_image(fig,
             err_message += """
 Exporting to EPS format requires the poppler library.  You can install
 poppler on MacOS or Linux with:
- 
+
     $ conda install poppler
-    
+
 Or, you can install it on MacOS using homebrew with:
 
     $ brew install poppler
@@ -1498,7 +1498,7 @@ def write_image(fig,
 For example:
 
     >>> import plotly.io as pio
-    >>> pio.write_image(fig, file_path, format='png') 
+    >>> pio.write_image(fig, file_path, format='png')
 """.format(file=file))
 
     # Request image

diff --git a/plotly/io/_templates.py b/plotly/io/_templates.py
@@ -1,10 +1,4 @@
 from __future__ import absolute_import
-from plotly.basedatatypes import BaseFigure
-from plotly.graph_objs import Figure
-from plotly.validators.layout import TemplateValidator
-from plotly.graph_objs.layout import Template
-from _plotly_utils.basevalidators import (
-    CompoundValidator, CompoundArrayValidator, is_array)
 
 import textwrap
 import pkgutil
@@ -44,7 +38,7 @@ def __init__(self):
         for template_name in default_templates:
             self._templates[template_name] = Lazy
 
-        self._validator = TemplateValidator()
+        self._validator = None
         self._default = None
 
     # ### Magic methods ###
@@ -61,6 +55,8 @@ def __iter__(self):
     def __getitem__(self, item):
         template = self._templates[item]
         if template is Lazy:
+            from plotly.graph_objs.layout._template import Template
+
             # Load template from package data
             path = os.path.join('package_data', 'templates', item + '.json')
             template_str = pkgutil.get_data('plotly', path).decode('utf-8')
@@ -81,6 +77,11 @@ def __delitem__(self, key):
         if self._default == key:
             self._default = None
 
+    def _validate(self, template):
+        if not self._validator:
+            from plotly.validators.layout import TemplateValidator
+            self._validator = TemplateValidator()
+
     def keys(self):
         return self._templates.keys()
 
@@ -189,6 +190,7 @@ def merge_templates(self, *args):
         if args:
             return reduce(self._merge_2_templates, args)
         else:
+            from plotly.graph_objs.layout._template import Template
             return Template()
 
     def _merge_2_templates(self, template1, template2):
@@ -250,6 +252,9 @@ def walk_push_to_template(fig_obj, template_obj, skip):
     skip: set of str
         Set of names of properties to skip
     """
+    from _plotly_utils.basevalidators import (
+        CompoundValidator, CompoundArrayValidator, is_array)
+
     for prop in list(fig_obj._props):
         if prop == 'template' or prop in skip:
             # Avoid infinite recursion
@@ -393,6 +398,8 @@ def to_templated(fig, skip=('title', 'text')):
     """
 
     # process fig
+    from plotly.basedatatypes import BaseFigure
+    from plotly.graph_objs import Figure
     if not isinstance(fig, BaseFigure):
         fig = Figure(fig)
 

diff --git a/plotly/io/_utils.py b/plotly/io/_utils.py
@@ -1,17 +1,15 @@
 from __future__ import absolute_import
 
-import plotly
-from plotly.basedatatypes import BaseFigure
-import plotly.graph_objs as go
-
 
 def validate_coerce_fig_to_dict(fig, validate):
+    from plotly.basedatatypes import BaseFigure
+    from plotly.graph_objs import Figure
     if isinstance(fig, BaseFigure):
         fig_dict = fig.to_dict()
     elif isinstance(fig, dict):
         if validate:
             # This will raise an exception if fig is not a valid plotly figure
-            fig_dict = plotly.graph_objs.Figure(fig).to_plotly_json()
+            fig_dict = Figure(fig).to_plotly_json()
         else:
             fig_dict = fig
     else:
@@ -22,6 +20,8 @@ def validate_coerce_fig_to_dict(fig, validate):
 
 
 def validate_coerce_output_type(output_type):
+    import plotly.graph_objs as go
+
     if output_type == 'Figure' or output_type == go.Figure:
         cls = go.Figure
     elif (output_type == 'FigureWidget' or
@@ -31,4 +31,4 @@ def validate_coerce_output_type(output_type):
         raise ValueError("""
 Invalid output type: {output_type}
     Must be one of: 'Figure', 'FigureWidget'""")
-    return cls
+    return cls