### 使用 Python 的 `sqlparse` 库解析 SQL 语句中的 WHERE 和 ON 条件 #### 解析 WHERE 子句 为了提取 SQL 语句中 WHERE 子句的值，可以使用 `sqlparse` 库。此库提供了强大的功能来解析和处理复杂的 SQL 查询。 下面是一段示例代码，展示如何获取并打印出 WHERE 子句中的条件： ```python import sqlparse def extract_where_conditions(sql_statement): parsed = sqlparse.parse(sql_statement)[0] tokens = list(parsed.flatten()) where_index = next((i for i, t in enumerate(tokens) if str(t).upper() == 'WHERE'), None) if where_index is not None: where_tokens = [] for token in tokens[where_index:]: if str(token).strip(): where_tokens.append(str(token)) # Join the tokens to form a string of conditions. where_condition_string = ''.join(where_tokens) return where_condition_string.strip() else: return "No WHERE clause found" # Example usage sql_query_with_where = "SELECT * FROM table_name WHERE column1 = 'value1' AND column2 > 10;" print(extract_where_conditions(sql_query_with_where))[^1] ``` 这段代码遍历了所有的标记（tokens），找到了关键字 “WHERE”，然后继续收集后续的所有有效标记直到结束，最后返回这些标记组成的字符串表示形式。 #### 解析 ON 子句 对于 JOIN 中的 ON 子句，同样可以通过查找特定的关键字来进行定位，并进一步提取其后的表达式作为 ON 条件的内容。 这里有一个类似的例子，用于捕获 ON 关键词之后的部分： ```python def extract_on_conditions(sql_statement): parsed = sqlparse.parse(sql_statement)[0] tokens = list(parsed.flatten()) on_index = next((i for i, t in enumerate(tokens) if str(t).upper().startswith('ON')), None) if on_index is not None: on_tokens = [] for token in tokens[on_index + 1:]: # Start after 'ON' if str(token).strip(): on_tokens.append(str(token)) # Join the tokens to form a string of conditions. on_condition_string = ''.join(on_tokens) return on_condition_string.strip() else: return "No ON clause found" # Example usage with an INNER JOIN that has an ON condition sql_query_with_join_and_on = """ SELECT * FROM first_table ft INNER JOIN second_table st ON ft.id = st.foreign_id; """ print(extract_on_conditions(sql_query_with_join_and_on))[^2] ``` 上述方法通过寻找关键词 ‘ON’ 并读取紧随其后的所有非空白字符序列来构建 ON 子句的结果串。