add env lookup for AADC_DB username and pass

SNflows · Feb 8, 2023 · 996a992 · 996a992
1 parent c86499a
commit 996a992
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/flows/aadc_db.py b/flows/aadc_db.py
@@ -9,10 +9,11 @@
 
 .. codeauthor:: Rasmus Handberg <rasmush@phys.au.dk>
 """
+import getpass
+import os
 
 import psycopg2 as psql
 from psycopg2.extras import DictCursor
-import getpass
 from tendrils.utils import load_config
 
 
@@ -41,15 +42,15 @@ def __init__(self, username=None, password=None):
         config = load_config()
 
         if username is None:
-            username = config.get('database', 'username', fallback=None)
+            username = config.get('database', 'username', fallback=os.environ.get("AUDBUsername", None))
             if username is None:
                 default_username = getpass.getuser()
                 username = input('Username [%s]: ' % default_username)
                 if username == '':
                     username = default_username
 
         if password is None:
-            password = config.get('database', 'password', fallback=None)
+            password = config.get('database', 'password', fallback=os.environ.get("AUDBPassword", None))
             if password is None:
                 password = getpass.getpass('Password: ')