过去几周我正在学习 PyParsing。我打算用它从 SQL 语句中获取表名。
我看过http://pyparsing.wikispaces.com/file/view/simpleSQL.py http://pyparsing.wikispaces.com/file/view/simpleSQL.py。但我打算保持语法简单，因为我并不试图解析 select 语句的每个部分，而是只查找表名。此外，为任何商业可用的现代数据库（如 Teradata）定义完整的语法也非常复杂。
#!/usr/bin/env python
from pyparsing import *
import sys
semicolon = Combine(Literal(';') + lineEnd)
comma = Literal(',')
lparen = Literal('(')
rparen = Literal(')')
# Keyword definition
update_kw, volatile_kw, create_kw, table_kw, as_kw, from_kw, \
where_kw, join_kw, left_kw, right_kw, cross_kw, outer_kw, \
on_kw , insert_kw , into_kw= \
map(lambda x: Keyword(x, caseless=True), \
['UPDATE', 'VOLATILE', 'CREATE', 'TABLE', 'AS', 'FROM',
'WHERE', 'JOIN' , 'LEFT', 'RIGHT' , \
'CROSS', 'OUTER', 'ON', 'INSERT', 'INTO'])
# Teradata SQL allows SELECT and well as SEL keyword
select_kw = Keyword('SELECT', caseless=True) | Keyword('SEL' , caseless=True)
# list of reserved keywords
reserved_words = (update_kw | volatile_kw | create_kw | table_kw | as_kw |
select_kw | from_kw | where_kw | join_kw |
left_kw | right_kw | cross_kw | on_kw | insert_kw |
into_kw)
# Identifier can be used as table or column names. They can't be reserved words
ident = ~reserved_words + Word(alphas, alphanums + '_')
# Recursive definition for table
table = Forward()
# simple table name can be identifer or qualified identifier e.g. schema.table
simple_table = Combine(Optional(ident + Literal('.')) + ident)
# table name can also a complete select statement used as table
nested_table = lparen.suppress() + select_kw.suppress() + SkipTo(from_kw).suppress() + \
from_kw.suppress() + table + rparen.suppress()
# table can be simple table or nested table
table << (nested_table | simple_table)
# comma delimited list of tables
table_list = delimitedList(table)
# Building from clause only because table name(s) will always appears after that
from_clause = from_kw.suppress() + table_list
txt = """
SELECT p, (SELECT * FROM foo),e FROM a, d, (SELECT * FROM z), b
"""
for token, start, end in from_clause.scanString(txt):
print token
这里值得一提的是。我使用“SkipTo(from_kw)”来跳过 SQL 语句中的列列表。这主要是为了避免为列列表定义语法，列列表可以是逗号分隔的标识符列表、许多函数名称、DW 分析函数等等。通过这个语法，我可以解析上面的语句以及 SELECT 列列表或表列表中的任何级别的嵌套。
['foo']
['a', 'd', 'z', 'b']
当 SELECT 有 where 子句时，我遇到问题：
nested_table = lparen.suppress() + select_kw.suppress() + SkipTo(from_kw).suppress() + \
from_kw.suppress() + table + rparen.suppress()
当 WHERE 子句存在时，相同的语句可能如下所示：
SELECT ... FROM a,d , (SELECT * FROM z WHERE (c1 = 1) and (c2 = 3)), p
我想将“nested_table”定义更改为：
nested_table = lparen.suppress() + select_kw.suppress() + SkipTo(from_kw).suppress() + \
from_kw.suppress() + table + Optional(where_kw + SkipTo(rparen)) + rparen
但这不起作用，因为它与“c = 1”后面的右括号匹配。我想知道的是如何跳到与“SELECT * FROM z...”之前的左括号匹配的右括号，我不知道如何使用 PyParsing 来做到这一点
另外，我还寻求一些建议，以了解从复杂的嵌套 SQL 中获取表名的最佳方法。非常感谢任何帮助。
谢谢
阿比吉特
