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

TechSynapse發表於2024-08-09
PythonSQL

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語句中包含註釋或字串字面量時。在實際應用中，您可能需要進一步調整正規表示式以處理這些情況。

相關文章