Python?提取出SQL語句中Where的值兩種方法(示例代碼)

更新時間：2024年08月10日 09:01:28 作者：TS86

為了提取SQL語句中WHERE子句的值,我們可以利用Python的sqlparse庫,這是一個專門用于解析SQL語句的庫,這篇文章主要介紹了Python?提取出SQL語句中Where的值的方法,需要的朋友可以參考下

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

為了提取SQL語句中WHERE子句的值，我們可以利用Python的sqlparse庫，這是一個專門用于解析SQL語句的庫。以下是一個示例代碼，演示如何使用sqlparse來提取WHERE子句中的條件。

首先，確保安裝了sqlparse庫。如果未安裝，可以使用pip安裝：

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子句中的所有值（特別是當它們包含復雜表達式或嵌套時）可能非常具有挑戰(zhàn)性。

下面，我將提供一個簡單的示例，該示例能夠處理一些基本的SQL查詢，并嘗試提取WHERE子句中的條件。請注意，這個示例可能無法處理所有可能的SQL查詢情況，特別是那些包含復雜邏輯或嵌套查詢的查詢。

import re  
def extract_where_clause(sql):  
    # 使用正則表達式匹配WHERE子句  
    # 這個正則表達式假設WHERE子句在SQL語句中直接跟在SELECT, UPDATE, DELETE等之后  
    # 并且可能包含空格、換行符等  
    # 注意：這個正則表達式非?；A，可能無法處理所有情況  
    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子句。這個示例將包括一個函數來執(zhí)行提取操作，并在腳本的末尾調用這個函數來測試幾個不同的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語句中包含注釋或字符串字面量時。在實際應用中，您可能需要進一步調整正則表達式以處理這些情況。

到此這篇關于Python 提取出SQL語句中Where的值的方法的文章就介紹到這了,更多相關Python 提取Where的值內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: