使用 sqlparse 库解析 SQL 查询语句

以下是一个简单的 SQL 解析器的 Python 代码示例,这个代码使用了 Python 的 SQL 解析库 sqlparse，它实现了一个完整的 SQL 解析器。它接受一个 SQL 查询语句作为输入，并返回一个字典，包含了查询类型和查询目标。具体来说，它使用 sqlparse.parse 函数解析 SQL 查询语句，并根据查询类型调用相应的解析函数来获取查询目标。目前，它支持解析 SELECT、INSERT、UPDATE 和 DELETE 四种类型的查询语句。对于 SELECT 查询语句，它将返回一个包含字段列表、表名和条件的字典；对于 INSERT 查询语句，它将返回一个包含表名、字段列表和值列表的字典；对于 UPDATE 查询语句，它将返回一个包含表名、字段列表、值列表和条件的字典；对于 DELETE 查询语句，它将返回一个包含表名和条件的字典。请注意，这个解析器可能无法处理所有的 SQL 查询语句，因此您可能需要根据实际情况进行修改和扩展。

import sqlparse

def parse_sql(sql):
    # 使用 sqlparse 库解析 SQL 查询语句
    parsed = sqlparse.parse(sql)[0]

    # 获取查询类型（SELECT、INSERT、UPDATE 或 DELETE）
    query_type = parsed.get_type()

    # 获取查询目标（表名、字段列表、值列表等）
    if query_type == 'SELECT':
        target = parse_select(parsed)
    elif query_type == 'INSERT':
        target = parse_insert(parsed)
    elif query_type == 'UPDATE':
        target = parse_update(parsed)
    elif query_type == 'DELETE':
        target = parse_delete(parsed)
    else:
        target = None

    return {'type': query_type, 'target': target}

def parse_select(parsed):
    # 获取字段列表
    fields = []
    for token in parsed.tokens:
        if isinstance(token, sqlparse.sql.IdentifierList):
            for identifier in token.get_identifiers():
                fields.append(identifier.value)
        elif isinstance(token, sqlparse.sql.Identifier):
            fields.append(token.value)

    # 获取表名
    for token in parsed.tokens:
        if isinstance(token, sqlparse.sql.Where):
            table = token.tokens[2].value
            break
    else:
        table = None

    # 获取条件
    for token in parsed.tokens:
        if isinstance(token, sqlparse.sql.Where):
            condition = token.tokens[4].value
            break
    else:
        condition = None

    return {'fields': fields, 'table': table, 'condition': condition}

def parse_insert(parsed):
    # 获取表名
    table = parsed.tokens[2].value

    # 获取字段列表
    fields = []
    for token in parsed.tokens:
        if isinstance(token, sqlparse.sql.Parenthesis):
            for identifier in token.get_identifiers():
                fields.append(identifier.value)

    # 获取值列表
    values = []
    for token in parsed.tokens:
        if isinstance(token, sqlparse.sql.Values):
            for value_list in token.get_parameters():
                values.append([value.value for value in value_list])

    return {'table': table, 'fields': fields, 'values': values}

def parse_update(parsed):
    # 获取表名
    table = parsed.tokens[2].value

    # 获取字段列表和值列表
    fields = []
    values = []
    for token in parsed.tokens:
        if isinstance(token, sqlparse.sql.IdentifierList):
            for identifier in token.get_identifiers():
                fields.append(identifier.value)
        elif isinstance(token, sqlparse.sql.Assignment):
            fields.append(token.left.value)
            values.append(token.right.value)

    # 获取条件
    for token in parsed.tokens:
        if isinstance(token, sqlparse.sql.Where):
            condition = token.tokens[4].value
            break
    else:
        condition = None

    return {'table': table, 'fields': fields, 'values': values, 'condition': condition}

def parse_delete(parsed):
    # 获取表名
    table = parsed.tokens[2].value

    # 获取条件
    for token in parsed.tokens:
        if isinstance(token, sqlparse.sql.Where):
            condition = token.tokens[4].value
            break
    else:
        condition = None

    return {'table': table, 'condition': condition}