expand build.py CLI to simplify action's code

Author: Massimiliano Pippi

.github/workflows/docs.yaml

@@ -80,9 +80,8 @@ jobs:
       - name: Publish docs
         # determine docs version for the commit pushed and publish accordingly using Mike
         if: github.event_name == 'push'
-        env:
-          REMOTE: https://x-access-token:${{secrets.GITHUB_TOKEN}}@github.com/${{github.repository}}.git
         run: |
           git config --global user.email "bot@arduino.cc"
           git config --global user.name "ArduinoBot"
-          python docs/build.py
+          git remote add upstream https://${{ github.actor }}:${{ secrets.GITHUB_TOKEN }}@github.com/${{ github.repository }}.git
+          python docs/build.py --remote upstream

docs/build.py

@@ -18,6 +18,7 @@
 import unittest
 import subprocess
 
+import click
 from git import Repo
 
 
@@ -75,11 +76,19 @@ def get_rel_branch_names(blist):
     return sorted(names, key=lambda x: int(x.split(".")[1]), reverse=True)
 
 
-def main(repo_dir):
-    # Git remote must be set to publish docs
-    remote = os.environ.get("REMOTE")
-    if not remote:
-        print("REMOTE env var must be set to publish, running dry mode")
+@click.command()
+@click.option("--test", is_flag=True)
+@click.option("--dry", is_flag=True)
+@click.option("--remote", default="origin", help="The git remote where to push.")
+def main(test, dry, remote):
+    # Run tests if requested
+    if test:
+        unittest.main(argv=[""], exit=False)
+        sys.exit(0)
+
+    # Detect repo root folder
+    here = os.path.dirname(os.path.realpath(__file__))
+    repo_dir = os.path.join(here, "..")
 
     # Get current repo
     repo = Repo(repo_dir)
@@ -102,12 +111,12 @@ def main(repo_dir):
         f"DOCS_VERSION={docs_version}",
         f"DOCS_ALIAS={alias}",
     ]
-    if remote:
-        subprocess.run(args, shell=True, check=True, cwd=repo_dir)
-    else:
+
+    if dry:
         print(" ".join(args))
+        return 0
 
-    return 0
+    subprocess.run(args, shell=True, check=True, cwd=repo_dir)
 
 
 # Usage:
@@ -119,9 +128,4 @@ def main(repo_dir):
 #         $python build.py
 #
 if __name__ == "__main__":
-    if len(sys.argv) > 1 and sys.argv[1] == "test":
-        unittest.main(argv=[""], exit=False)
-        sys.exit(0)
-
-    here = os.path.dirname(os.path.realpath(__file__))
-    sys.exit(main(os.path.join(here, "..")))
+    sys.exit(main())

requirements_docs.txt

@@ -1,4 +1,5 @@
 mkdocs<1.2
 mkdocs-material<5
 mike
-gitpython
+gitpython
+click<7.2