Python 提取出SQL语句中Where的值的方法

1.方法一：使用sqlparse库的方法

为了提取SQL语句中WHERE子句的值，我们可以利用Python的sqlparse库，这是一个专门用于解析SQL语句的库。以下是一个示例代码，演示如何使用sqlparse来提取WHERE子句中的条件。

首先，确保安装了sqlparse库。如果未安装，可以使用pip安装：

bash复制代码

pip install sqlparse

然后，我们可以编写以下Python代码来提取WHERE子句的值：

import sqlparse
from sqlparse.sql import IdentifierList, Identifier
from sqlparse.tokens import Keyword, DML  

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

    # 提取WHERE子句
    where_seen = False
    for item in parsed.tokens:
        if where_seen:
            if is_subselect(item):
                where_seen = False
            else:
                # 这里的item可能是WHERE子句的一部分
                print(item)
        elif item.ttype is Keyword and item.value.upper() == 'WHERE':
            where_seen = True  

def is_subselect(parsed):
    if not parsed.is_group:
        return False
    for item in parsed.tokens:
        if item.ttype is DML and item.value.upper() == 'SELECT':
            return True
    return False  

# 示例SQL语句
sql = """
SELECT * FROM users
WHERE id = 10 AND status = 'active' OR name = 'John Doe';
"""  

extract_where_values(sql)

在这个例子中，extract_where_values函数接收一个SQL语句作为输入，然后使用sqlparse解析它。它遍历解析后的语句的标记（tokens），寻找WHERE关键字。一旦找到，它将打印出WHERE子句中的所有内容，直到遇到另一个子查询或SQL语句的结尾。

这个代码展示了如何提取和识别SQL语句中的WHERE子句。在实际应用中，我们可能需要更复杂的逻辑来处理更复杂的SQL语句，包括嵌套查询、复杂的条件表达式等。

2.方法二：使用正则表达式

要从SQL语句中提取WHERE子句的值，我们可以使用Python的正则表达式（re模块）来匹配和提取这些值。但是，需要注意的是，SQL语句的结构可能非常复杂，包含嵌套查询、子查询、函数、操作符等，因此完全准确地提取WHERE子句中的所有值（特别是当它们包含复杂表达式或嵌套时）可能非常具有挑战性。

下面，我将提供一个简单的示例，该示例能够处理一些基本的SQL查询，并尝试提取WHERE子句中的条件。请注意，这个示例可能无法处理所有可能的SQL查询情况，特别是那些包含复杂逻辑或嵌套查询的查询。

import re  

def extract_where_clause(sql):
    # 使用正则表达式匹配WHERE子句
    # 这个正则表达式假设WHERE子句在SQL语句中直接跟在SELECT, UPDATE, DELETE等之后
    # 并且可能包含空格、换行符等
    # 注意：这个正则表达式非常基础，可能无法处理所有情况
    pattern = r'(?<=WHERE\s+)(.*?)(?=\s*(?:ORDER BY|GROUP BY|LIMIT|;|$))'
    match = re.search(pattern, sql, re.IGNORECASE | re.DOTALL)  

    if match:
        return match.group(0).strip()
    else:
        return "No WHERE clause found."  

# 示例SQL语句
sql_examples = [
    "SELECT * FROM users WHERE id = 10 AND name = 'John';",
    "UPDATE users SET status = 'active' WHERE age > 30 AND status = 'inactive';",
    "DELETE FROM orders WHERE order_date < '2023-01-01';",
    "SELECT * FROM products;",  # 没有WHERE子句
    "SELECT * FROM products WHERE (price > 100 OR quantity < 10) AND category = 'Electronics';"
]  

# 遍历示例并打印结果
for sql in sql_examples:
    print(f"Original SQL: {sql}")
    print(f"Extracted WHERE Clause: {extract_where_clause(sql)}\n")

说明：

（1）正则表达式：这个正则表达式尝试匹配WHERE关键字后直到遇到ORDER BY、GROUP BY、LIMIT、语句结束符（;）或字符串末尾的任意字符序列。它使用了re.IGNORECASE来忽略大小写，re.DOTALL来允许.匹配包括换行符在内的任意字符。

（2）限制：这个正则表达式假设WHERE子句是直接跟在SQL语句的主要操作（如SELECT, UPDATE, DELETE）之后的，并且WHERE子句之后直接跟着的是其他SQL子句或语句结束符。这在一些复杂的SQL语句中可能不成立，特别是当WHERE子句被嵌套在子查询中时。

（3）输出：对于每个示例SQL语句，代码将打印出原始SQL语句和提取的WHERE子句（如果存在）。

这个示例提供了一个基本的起点，但根据具体需求，您可能需要调整正则表达式或采用更复杂的解析方法（如使用SQL解析库）来处理更复杂的SQL查询。

接下来，我将提供一个更具体的代码示例，并给出一个完整的Python脚本，该脚本使用正则表达式来提取SQL语句中的WHERE子句。这个示例将包括一个函数来执行提取操作，并在脚本的末尾调用这个函数来测试几个不同的SQL语句。

请注意，这个示例仍然基于正则表达式，并且可能无法处理所有复杂的SQL查询情况。对于更复杂的SQL解析，您可能需要考虑使用专门的SQL解析库，例如上文提到的sqlparse库的方法。

import re  

def extract_where_clause(sql):
    """
    从SQL语句中提取WHERE子句的内容。  

    参数:
    sql (str): SQL查询语句。  

    返回:
    str: 提取的WHERE子句内容（如果存在），否则返回"No WHERE clause found."。
    """
    # 使用正则表达式匹配WHERE子句
    # 这个正则表达式尝试匹配WHERE关键字后直到遇到SQL语句结束或特定SQL子句开始的位置
    pattern = r'(?<=WHERE\s+)(.*?)(?=\s*(?:ORDER BY|GROUP BY|LIMIT|;|$))'
    match = re.search(pattern, sql, re.IGNORECASE | re.DOTALL)  

    if match:
        return match.group(0).strip()
    else:
        return "No WHERE clause found."  

# 完整的Python脚本
if __name__ == "__main__":
    # 示例SQL语句
    sql_examples = [
        "SELECT * FROM users WHERE id = 10 AND name = 'John';",
        "UPDATE users SET status = 'active' WHERE age > 30 AND status = 'inactive';",
        "DELETE FROM orders WHERE order_date < '2023-01-01';",
        "SELECT * FROM products;",  # 没有WHERE子句
        "SELECT * FROM products WHERE (price > 100 OR quantity < 10) AND category = 'Electronics';",
        "SELECT * FROM (SELECT * FROM nested WHERE nested_id = 1) AS subquery WHERE subquery.id = 5;"  # 嵌套查询
    ]  

    # 遍历示例并打印结果
    for sql in sql_examples:
        print(f"Original SQL: {sql}")
        where_clause = extract_where_clause(sql)
        print(f"Extracted WHERE Clause: {where_clause}\n")  

# 输出将显示每个SQL语句的原始形式和提取的WHERE子句（如果存在）

在这个示例中，extract_where_clause函数使用了一个正则表达式来查找WHERE关键字后的内容，直到遇到ORDER BY、GROUP BY、LIMIT、SQL语句的结束（;）或字符串的末尾。然后，它返回匹配到的内容（如果有的话），否则返回一个说明没有找到WHERE子句的消息。

请注意，对于包含嵌套查询的SQL语句（如示例中的最后一个），这个正则表达式可能无法正确提取嵌套查询内部的WHERE子句，因为它只查找最外层的WHERE子句。要处理这种情况，您可能需要编写更复杂的正则表达式或使用SQL解析库。

此外，这个示例中的正则表达式使用了re.DOTALL标志，允许.匹配包括换行符在内的任意字符，这对于处理跨越多行的SQL语句很有用。然而，这也可能导致在不应该匹配的地方进行匹配，特别是当SQL语句中包含注释或字符串字面量时。在实际应用中，您可能需要进一步调整正则表达式以处理这些情况。