Python中使用SQLparse库解析SQL语句并提取表名和字段名

SQLparse是一个用于解析SQL语句的Python库。它可以将SQL语句解析成结构化的Python对象，并提取出其中的表名和字段名。下面是一个使用SQLparse库解析SQL语句并提取表名和字段名的示例：

import sqlparse

def extract_tables(sql):
    statements = sqlparse.parse(sql)
    tables = []

    for statement in statements:
        for token in statement.tokens:
            if isinstance(token, sqlparse.sql.IdentifierList):
                for identifier in token.get_identifiers():
                    tables.append(identifier.get_real_name())
            elif isinstance(token, sqlparse.sql.Identifier):
                tables.append(token.get_real_name())

    return tables

def extract_columns(sql):
    statements = sqlparse.parse(sql)
    columns = []

    for statement in statements:
        for token in statement.tokens:
            if isinstance(token, sqlparse.sql.IdentifierList):
                for identifier in token.get_identifiers():
                    columns.append(identifier.get_real_name())
            elif isinstance(token, sqlparse.sql.Identifier):
                columns.append(token.get_real_name())

    return columns

# 示例SQL语句
sql = """
SELECT column1, column2
FROM table1
WHERE column3 = 'value'
"""

# 提取表名
tables = extract_tables(sql)
print("Tables:", tables)

# 提取字段名
columns = extract_columns(sql)
print("Columns:", columns)

上述示例代码首先导入了sqlparse库，并定义了两个函数extract_tables和extract_columns，用于分别提取表名和字段名。

extract_tables函数使用SQLparse库将SQL语句解析成结构化的Python对象，然后遍历这个对象，提取其中的表名。在遍历的过程中，对于每个标识符列表（IdentifierList）或标识符（Identifier），通过调用get_real_name()方法获取表名的真实名称，并将其添加到列表中。

extract_columns函数与extract_tables函数类似，不同的是它提取的是字段名。同样地，在遍历的过程中，对于每个标识符列表（IdentifierList）或标识符（Identifier），通过调用get_real_name()方法获取字段名的真实名称，并将其添加到列表中。

在示例代码中，我们定义了一个SQL语句作为输入，并调用extract_tables和extract_columns函数来提取其中的表名和字段名。最后，我们将结果打印出来。

运行示例代码将输出以下结果：

Tables: ['table1']
Columns: ['column1', 'column2', 'column3']

从输出结果可以看出，我们成功地从SQL语句中提取出了表名和字段名。

总结来说，使用SQLparse库解析SQL语句并提取表名和字段名的步骤如下：首先使用sqlparse.parse()函数将SQL语句解析成结构化的Python对象，然后遍历这个对象，对于每个标识符列表（IdentifierList）或标识符（Identifier），通过调用get_real_name()方法获取表名或字段名的真实名称，并将其添加到列表中。