20
20
21
21
db_host = os .getenv ("DB_HOST" )
22
22
db_port = os .getenv ("DB_PORT" )
23
+ db_csv = os .getenv ("DB_CSV" )
24
+ db_orph = os .getenv ("DB_ORPH" )
23
25
db_user = os .getenv ("DB_USER" )
24
26
db_password = os .getenv ("DB_PASSWORD" )
25
27
26
28
27
29
def check_env_variables ():
28
30
required_env_vars = [
29
31
"GITHUB_TOKEN" , "DB_HOST" , "DB_PORT" ,
30
- "DB_USER" , "DB_PASSWORD" , "GITEA_TOKEN"
32
+ "DB_CSV" , "DB_ORPH" , " DB_USER" , "DB_PASSWORD" , "GITEA_TOKEN"
31
33
]
32
34
for var in required_env_vars :
33
35
if os .getenv (var ) is None :
@@ -471,7 +473,7 @@ def main(org, gh_org, rtctable, opentable, string):
471
473
check_env_variables ()
472
474
csv_erase ()
473
475
474
- conn_csv = connect_to_db ("csv" )
476
+ conn_csv = connect_to_db (db_csv )
475
477
cur_csv = conn_csv .cursor ()
476
478
477
479
g = Github (github_token )
@@ -492,7 +494,7 @@ def main(org, gh_org, rtctable, opentable, string):
492
494
update_service_titles (cur_csv , rtctable )
493
495
add_squad_column (cur_csv , rtctable )
494
496
495
- conn_orph = connect_to_db ("orph" )
497
+ conn_orph = connect_to_db (db_orph )
496
498
cur_orph = conn_orph .cursor ()
497
499
498
500
cur_orph .execute (f"DROP TABLE IF EXISTS { opentable } " )
0 commit comments