github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-01 17:16:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the issue when the MySQL table name had the keyword in it (#358)

Browse Source 
When a table name with special characters, such as trigger, exists in
the database, the following error is reported.
```log
2023-07-25 00:02:47 | ERROR | stderr | sqlalchemy.exc.ProgrammingError: (pymysql.err.ProgrammingError) (1064, "You have an error in your SQL syntax; check the manual that corresponds to your MySQL server version for the right syntax to use near 'trigger' at line 1")
2023-07-25 00:02:47 | ERROR | stderr | [SQL: SHOW INDEXES FROM trigger]
```
This commit is contained in:
magic.chen
2023-07-25 00:23:47 +08:00
committed by GitHub
parent aaccebcd03 34aa73fa76
commit 7233b9711f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/common/sql_database.py
View File

@@ -420,14 +420,14 @@ class Database:
def get_indexes(self, table_name):
"""Get table indexes about specified table."""
session = self._db_sessions()
cursor = session.execute(text(f"SHOW INDEXES FROM {table_name}"))
cursor = session.execute(text(f"SHOW INDEXES FROM `{table_name}`"))
indexes = cursor.fetchall()
return [(index[2], index[4]) for index in indexes]
def get_show_create_table(self, table_name):
"""Get table show create table about specified table."""
session = self._db_sessions()
cursor = session.execute(text(f"SHOW CREATE TABLE {table_name}"))
cursor = session.execute(text(f"SHOW CREATE TABLE `{table_name}`"))
ans = cursor.fetchall()
res = ans[0][1]
res = re.sub(r"\s*ENGINE\s*=\s*InnoDB\s*", " ", res, flags=re.IGNORECASE)