github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-06 02:46:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

web bug fix

Browse Source
This commit is contained in:
yhjun1026 2023-07-24 17:13:27 +08:00
parent 94fb112979
commit ea97bfccc7
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
pilot/connections/rdbms/rdbms_connect.py
View File

@ -74,6 +74,21 @@ class RDBMSDatabase(BaseConnect):
self._sample_rows_in_table_info = set() self._sample_rows_in_table_info = set()
self._indexes_in_table_info = indexes_in_table_info self._indexes_in_table_info = indexes_in_table_info
self._metadata = MetaData()
self._metadata.reflect(bind=self._engine)
# including view support by adding the views as well as tables to the all
# tables list if view_support is True
self._all_tables = set(
self._inspector.get_table_names(schema=self._engine.url.database)
+ (
self._inspector.get_view_names(schema=self._engine.url.database)
if self.view_support
else []
)
)
@classmethod @classmethod
def from_uri_db( def from_uri_db(
cls, cls,
@ -121,15 +136,11 @@ class RDBMSDatabase(BaseConnect):
def get_table_names(self) -> Iterable[str]: def get_table_names(self) -> Iterable[str]:
"""Get names of tables available.""" """Get names of tables available."""
warnings.warn(
"This method is deprecated - please use `get_usable_table_names`."
)
return self.get_usable_table_names() return self.get_usable_table_names()
def get_session(self): def get_session(self):
session = self._db_sessions() session = self._db_sessions()
self._metadata = MetaData()
self._metadata.reflect(bind=self._engine)
return session return session