Add --local flag to updateplotlyjsdev

Author: ayjayt

packages/python/plotly/setup.py

@@ -1,7 +1,9 @@
 import os
 import sys
+import time
 import platform
 import json
+import shutil
 
 from setuptools import setup, Command
 from setuptools.command.egg_info import egg_info
@@ -137,10 +139,10 @@ class NPM(Command):
     ]
 
     def initialize_options(self):
-        pass
+        self.local = None
 
     def finalize_options(self):
-        pass
+        self.set_undefined_options("updateplotlyjsdev", ("local", "local"))
 
     def get_npm_name(self):
         npmName = "npm"
@@ -187,6 +189,14 @@ def run(self):
                 stdout=sys.stdout,
                 stderr=sys.stderr,
             )
+            if self.local is not None:
+                plotly_archive = os.path.join(self.local, "plotly.js.tgz")
+                check_call(
+                    [npmName, "install", plotly_archive],
+                    cwd=node_root,
+                    stdout=sys.stdout,
+                    stderr=sys.stderr,
+                )
             check_call(
                 [npmName, "run", "build:prod"],
                 cwd=node_root,
@@ -224,6 +234,9 @@ def run(self):
 
         perform_codegen()
 
+def overwrite_schema_local(uri):
+    path = os.path.join(here, "codegen", "resources", "plot-schema.json")
+    shutil.copyfile(uri, path)
 
 def overwrite_schema(url):
     import requests
@@ -234,6 +247,9 @@ def overwrite_schema(url):
     with open(path, "wb") as f:
         f.write(req.content)
 
+def overwrite_bundle_local(uri):
+    path = os.path.join(here, "plotly", "package_data", "plotly.min.js")
+    shutil.copyfile(uri, path)
 
 def overwrite_bundle(url):
     import requests
@@ -281,7 +297,6 @@ def request_json(url):
     req = requests.get(url)
     return json.loads(req.content.decode("utf-8"))
 
-
 def get_latest_publish_build_info(repo, branch):
 
     url = (
@@ -303,6 +318,11 @@ def get_latest_publish_build_info(repo, branch):
     # Extract build info
     return {p: build[p] for p in ["vcs_revision", "build_num", "committer_date"]}
 
+def get_bundle_schema_local(local):
+    plotly_archive = os.path.join(local, "plotly.js.tgz")
+    plotly_bundle = os.path.join(local, "dist/plotly.min.js")
+    plotly_schemas = os.path.join(local, "dist/plot-schema.json")
+    return plotly_archive, plotly_bundle, plotly_schemas
 
 def get_bundle_schema_urls(build_num):
     url = (
@@ -390,31 +410,45 @@ class UpdateBundleSchemaDevCommand(Command):
     def initialize_options(self):
         self.devrepo = None
         self.devbranch = None
+        self.local = None
 
     def finalize_options(self):
         self.set_undefined_options("updateplotlyjsdev", ("devrepo", "devrepo"))
         self.set_undefined_options("updateplotlyjsdev", ("devbranch", "devbranch"))
+        self.set_undefined_options("updateplotlyjsdev", ("local", "local"))
 
     def run(self):
-        build_info = get_latest_publish_build_info(self.devrepo, self.devbranch)
+        if self.local is None:
+            build_info = get_latest_publish_build_info(self.devrepo, self.devbranch)
 
-        archive_url, bundle_url, schema_url = get_bundle_schema_urls(
-            build_info["build_num"]
-        )
+            archive_url, bundle_url, schema_url = get_bundle_schema_urls(
+                build_info["build_num"]
+            )
+
+            # Update bundle in package data
+            overwrite_bundle(bundle_url)
 
-        # Update bundle in package data
-        overwrite_bundle(bundle_url)
+            # Update schema in package data
+            overwrite_schema(schema_url)
+        else:
+            # this info could be more informative but
+            # it doesn't seem as useful in a local context
+            # and requires dependencies and programming.
+            build_info = {"vcs_revision": "local", "committer_date": str(time.time())}
+            self.devrepo = self.local
+            self.devbranch = ""
 
-        # Update schema in package data
-        overwrite_schema(schema_url)
+            archive_uri, bundle_uri, schema_uri = get_bundle_schema_local(self.local)
+            overwrite_bundle_local(bundle_uri)
+            overwrite_schema_local(schema_uri)
 
         # Update plotly.js url in package.json
         package_json_path = os.path.join(node_root, "package.json")
         with open(package_json_path, "r") as f:
             package_json = json.load(f)
 
         # Replace version with bundle url
-        package_json["dependencies"]["plotly.js"] = archive_url
+        package_json["dependencies"]["plotly.js"] = archive_url if self.local is None else archive_uri
         with open(package_json_path, "w") as f:
             json.dump(package_json, f, indent=2)
 
@@ -430,11 +464,13 @@ class UpdatePlotlyJsDevCommand(Command):
     user_options = [
         ("devrepo=", None, "Repository name"),
         ("devbranch=", None, "branch or pull/number"),
+        ("local=", None, "local copy of repo, used by itself")
     ]
 
     def initialize_options(self):
         self.devrepo = "plotly/plotly.js"
         self.devbranch = "master"
+        self.local = None
 
     def finalize_options(self):
         pass