Add new toplevel updateplotlyjs setup.py command

 - Reads plotly.js version from js/package.json
 - Downloads plot schema
 - Downloads minified plotly.js bundle
 - Performs code generation

Author: jonmmease

js/package.json

@@ -29,7 +29,7 @@
     "ify-loader": "^1.1.0"
   },
   "dependencies": {
-    "plotly.js": "^1.38.2",
+    "plotly.js": "1.38.2",
     "@jupyter-widgets/base": "^1.0.0",
     "lodash": "^4.17.4"
   },

plotly/version.py

@@ -1 +1 @@
-__version__ = '3.0.0rc2'
+__version__ = '3.0.0rc5'

setup.py

@@ -8,8 +8,7 @@
 import os
 import sys
 import platform
-
-plotly_js_version = '1.38.2'
+import json
 
 exec(open('plotly/version.py').read())
 
@@ -22,6 +21,16 @@
                 os.environ.get('PATH', os.defpath),
 ])
 
+
+# Load plotly.js version from js/package.json
+def plotly_js_version():
+    with open('js/package.json', 'rt') as f:
+        package_json = json.load(f)
+        version = package_json['dependencies']['plotly.js']
+
+    return version
+
+
 def readme():
     with open('README.rst') as f:
         return f.read()
@@ -138,13 +147,14 @@ def finalize_options(self):
 
     def run(self):
         if sys.version_info.major != 3 or sys.version_info.minor < 6:
-            raise ImportError('Code generation must be executed with Python >= 3.6')
+            raise ImportError(
+                'Code generation must be executed with Python >= 3.6')
 
         from codegen import perform_codegen
         perform_codegen()
 
 
-class DownloadSchemaCommand(Command):
+class UpdateSchemaCommand(Command):
     description = 'Download latest version of the plot-schema JSON file'
     user_options = []
 
@@ -160,14 +170,14 @@ def run(self):
 
         import urllib.request
         url = ('https://raw.githubusercontent.com/plotly/plotly.js/'
-               'v%s/dist/plot-schema.json' % plotly_js_version)
+               'v%s/dist/plot-schema.json' % plotly_js_version())
         with urllib.request.urlopen(url) as response:
 
             with open('plotly/package_data/plot-schema.json', 'wb') as f:
                 f.write(response.read())
 
 
-class DownloadPlotlyJsCommand(Command):
+class UpdateBundleCommand(Command):
     description = 'Download latest version of the plot-schema JSON file'
     user_options = []
 
@@ -183,13 +193,29 @@ def run(self):
 
         import urllib.request
         url = ('https://raw.githubusercontent.com/plotly/plotly.js/'
-               'v%s/dist/plotly.min.js' % plotly_js_version)
+               'v%s/dist/plotly.min.js' % plotly_js_version())
         with urllib.request.urlopen(url) as response:
 
             with open('plotly/package_data/plotly.min.js', 'wb') as f:
                 f.write(response.read())
 
 
+class UpdatePlotlyJsCommand(Command):
+    description = 'Update project to a new version of plotly.js'
+    user_options = []
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        self.run_command('updatebundle')
+        self.run_command('updateschema')
+        self.run_command('codegen')
+
+
 graph_objs_packages = [
     d[0] for d in os.walk('plotly/graph_objs')
     if not d[0].endswith('__pycache__')]
@@ -260,7 +286,8 @@ def run(self):
           'sdist': js_prerelease(sdist, strict=True),
           'jsdeps': NPM,
           'codegen': CodegenCommand,
-          'updateschema': DownloadSchemaCommand,
-          'updateplotlyjs': DownloadPlotlyJsCommand
+          'updateschema': UpdateSchemaCommand,
+          'updatebundle': UpdateBundleCommand,
+          'updateplotlyjs': js_prerelease(UpdatePlotlyJsCommand)
       },
 )