Nicer error messages for dsn syntax errors (#195)

dalibo · Mar 17, 2021 · aeed63e · aeed63e
1 parent 95faf1e
commit aeed63e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pgactivity/data.py b/pgactivity/data.py
@@ -7,7 +7,7 @@
 import psutil
 import psycopg2
 import psycopg2.extras
-from psycopg2.errors import InterfaceError, InvalidPassword, OperationalError
+from psycopg2.errors import InterfaceError, InvalidPassword, OperationalError, ProgrammingError
 from psycopg2.extensions import connection
 
 from . import queries
@@ -395,6 +395,11 @@ def pg_connect(
                 raise SystemExit("pg_activity: FATAL: %s" % clean_str(msg))
             else:
                 raise Exception("Could not connect to PostgreSQL")
+        except ProgrammingError as err:
+            errmsg = str(err).strip()
+            if errmsg.startswith("invalid dsn"):
+                raise SystemExit(f"ERROR: {errmsg}\nPlease refer to the 'Connection Control Functions' section of the PostgreSQL documentation")
+            raise
         else:
             break
     return data