Update main.py

bcgov · May 15, 2024 · 0edc786 · 0edc786
1 parent e2f0c93
commit 0edc786
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -9,7 +9,7 @@
     ODS_USER = os.environ["ODS_USER"]
     ODS_PASSWORD = os.environ["ODS_PASSWORD"]
     ODS_DB = os.environ["ODS_DB"]
-    GH_TOKEN_SECRET = os.environ["GH_TOKEN_SECRET"]
+    G_SECRET = os.environ["G_SECRET"]
 
 except KeyError:
     SOME_SECRET = "Token not available!"
@@ -87,7 +87,7 @@ def create_dag_definition(application):
     #DAG = """{DAG}"""
     DAG = str(DAG)
 
-    g = Github(GITHUB_TOKEN_SECRET)
+    g = Github(G_SECRET)
     repo = g.get_repo('bcgov/nr-airflow')
 
     repo.create_file(f'dags/{dag_id}.py', 'upload dags', DAG, branch='ui-testing')
@@ -112,7 +112,7 @@ def extract_names(input_list):
 
 # Main function to orchestrate the process
 def main():
-    g = Github(GITHUB_TOKEN_SECRET)
+    g = Github(G_SECRET)
     repo = g.get_repo("bcgov/nr-airflow")
     contents = repo.get_contents("/dags", ref="ui-testing")
     existing_dags= [c.path for c in contents]