summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorIrina Truong <i.chernyavska@gmail.com>2017-03-13 11:12:09 -0700
committerIrina Truong <i.chernyavska@gmail.com>2017-03-13 11:12:09 -0700
commit6d37b9209d1f38fe6b45df2680181a8814e0075f (patch)
treee39683cbf21c4c7cbebd55e6b39583d1747685aa
parentdcd9cbac4f3492b674b414c9bc1ddc54cbf61c98 (diff)
Combine select queries for connection properties.
-rw-r--r--pgcli/pgexecute.py7
1 files changed, 3 insertions, 4 deletions
diff --git a/pgcli/pgexecute.py b/pgcli/pgexecute.py
index 451a0a89..20ebd258 100644
--- a/pgcli/pgexecute.py
+++ b/pgcli/pgexecute.py
@@ -189,10 +189,9 @@ class PGExecute(object):
# When we connect using a DSN, we don't really know what db,
# user, etc. we connected to. Let's read it.
# Note: moved this after setting autocommit because of #664.
- db = self._select_one(cursor, 'select current_database()')[0]
- user = self._select_one(cursor, 'select current_user')[0]
- host = self._select_one(cursor, 'select inet_server_addr()')[0]
- port = self._select_one(cursor, 'select inet_server_port()')[0]
+ db, user, host, port = self._select_one(
+ cursor,
+ 'select current_database(), current_user, inet_server_addr(), inet_server_port()')
self.dbname = db
self.user = user