Try-except case when pgpass does not exist

sdss · May 13, 2020 · 75b0455 · 75b0455
1 parent 737f610
commit 75b0455
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/sdssdb/connection.py b/python/sdssdb/connection.py
@@ -11,8 +11,8 @@
 import re
 import socket
 
+import pgpasslib
 import six
-from pgpasslib import getpass
 
 from sdssdb import _peewee, _sqla, config, log
 
@@ -363,7 +363,10 @@ def _conn(self, dbname, silent_on_fail=False, **params):
             """Connects to the DB and tests the connection."""
 
             if 'password' not in params:
-                params['password'] = getpass(dbname=dbname, **params)
+                try:
+                    params['password'] = pgpasslib.getpass(dbname=dbname, **params)
+                except pgpasslib.FileNotFound:
+                    params['password'] = None
 
             PostgresqlDatabase.init(self, dbname, **params)