More debug output

a4z · Dec 22, 2023 · b2d6f8e · b2d6f8e
1 parent 5556b53
commit b2d6f8e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -52,7 +52,7 @@ jobs:
         path: "requirements.txt"
     - name: Download Generator
       run: |
-        python get-generator.py $GENERATOR_BUILID
+        python get-generator.py $GENERATOR_BUILID ${{ secrets.GENERATOR_DL_TOKEN }}
       env:
         GENERATOR_BUILID: ${{ env.GENERATOR_BUILID }}
         GENERATOR_DL_TOKEN: ${{ secrets.GENERATOR_DL_TOKEN }}

diff --git a/get-generator.py b/get-generator.py
@@ -27,6 +27,17 @@
     10. Run the generator
 """
 
+dl_arg = sys.argv[1]
+if not dl_arg:
+    sys.exit('No argument given')
+
+
+key_arg = sys.argv[2]
+
+if key_arg:
+    print(f'Using key: {key_arg}')
+    os.environ['GENERATOR_DL_TOKEN'] = key_arg
+
 # Get the token from the environment variable
 TOKEN_ENV_VAR='GENERATOR_DL_TOKEN'
 TOKEN = os.getenv(TOKEN_ENV_VAR)
@@ -148,10 +159,6 @@ def get_latest_pre_release(owner, repo):
             make_executable(dl_file)
             break
 
-# Get the argument, TODO, use argparse and add optional a folder argument
-dl_arg = sys.argv[1]
-if not dl_arg:
-    sys.exit('No argument given')
 
 owner = 'cross-language-cpp'
 repo = 'djinni-generator'