python 使用sqlparse 中的 lexer模块解析表血缘 代码发我

以下是一个简单的示例，使用 sqlparse 的 lexer 模块解析 SQL 语句中的表血缘关系：

import sqlparse

sql = """
SELECT a.*, b.*
FROM table1 a
JOIN table2 b ON a.id = b.id
"""

parsed = sqlparse.parse(sql)[0]
tables = set()

# 遍历 SQL 语句中的每个令牌
for token in parsed.tokens:
    if token.ttype is None:
        # 跳过空令牌
        continue
    elif token.ttype in (sqlparse.tokens.Keyword, sqlparse.tokens.Keyword.DML):
        # 如果是关键字或 DML 语句，则清空已解析的表名
        tables.clear()
    elif token.ttype is sqlparse.tokens.Keyword.Table:
        # 如果是表名，则将其添加到表集合中
        tables.add(token.value)
    elif token.ttype is sqlparse.tokens.Punctuation:
        # 如果是逗号，则将前面的表名添加到后面的表的血缘关系中
        if token.value == ',':
            for table in tables:
                if table not in table_relations:
                    table_relations[table] = set()
                for other_table in tables:
                    if other_table != table:
                        table_relations[table].add(other_table)

print(table_relations)

注意，这只是一个简单的示例，无法处理复杂的 SQL 语句。如果你需要处理更复杂的场景，可能需要使用更高级的 SQL 解析器和处理器。