python - 从 sql 字符串中提取表名、列等参数

Question

我需要为 Python 3 中的给定 SQL 字符串提取表名、选定列和 WHERE 语句等参数。是否有一个库可以执行此操作，我可以为我的案例获取一个示例吗？像这样的东西:

sql = 'select cola, colb, colc from "someschema"."mytable" where id = 1'
result = sqlparse(sql)
print(result.columns)
>> ["cola","colb","colc"]

print(result.tables)
>> ["mytable"]

如果问题不清楚:之前针对C#问过同样的问题 extract column names and table name from any "SQL query" String

Answer 1

你可以使用sqlparser

提取表名的例子在这里:https://github.com/andialbrecht/sqlparse/blob/master/examples/extract_table_names.py

提取列的例子在这里: https://github.com/andialbrecht/sqlparse/blob/master/examples/column_defs_lowlevel.py