github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-31 15:47:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: add error message log out

Browse Source
This commit is contained in:
lozzow 2023-09-25 13:05:04 +00:00
parent eee2072acf
commit 6a03c24ee7
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pilot/connections/rdbms/conn_postgresql.py
View File

@ -60,10 +60,14 @@ class PostgreSQLDatabase(RDBMSDatabase):
def get_collation(self):
"""Get collation."""
session = self._db_sessions()
cursor = session.execute(text("SELECT datcollate AS collation FROM pg_database WHERE datname = current_database();"))
collation = cursor.fetchone()[0]
return collation
try:
session = self._db_sessions()
cursor = session.execute(text("SELECT datcollate AS collation FROM pg_database WHERE datname = current_database();"))
collation = cursor.fetchone()[0]
return collation
except Exception as e:
print("postgresql get collation error: ", e)
return None
def get_users(self):
"""Get user info."""
@ -72,6 +76,7 @@ class PostgreSQLDatabase(RDBMSDatabase):
users = cursor.fetchall()
return [user[0] for user in users]
except Exception as e:
print("postgresql get users error: ", e)
return []
def get_fields(self, table_name):