如何提取SQL查询中的所有表(包括where引用)？

import re

def extract_tables_from_sql(sql):
    # 解析SQL查询语句，提取表名和where条件
    table_pattern = r'(?:FROM|JOIN)\s+(\w+)'
    where_pattern = r'WHERE\s+(\w+)'

    tables = []
    
    # 提取表名
    table_matches = re.findall(table_pattern, sql, re.IGNORECASE)
    tables.extend(table_matches)
    
    # 提取where条件中的表名
    where_matches = re.findall(where_pattern, sql, re.IGNORECASE)
    tables.extend(where_matches)
    
    # 去重处理
    tables = list(set(tables))
    
    return tables

# 示例SQL查询语句
sql_query = "SELECT * FROM table1 JOIN table2 ON table1.id = table2.id WHERE table1.name = 'John'"

# 提取表名
tables = extract_tables_from_sql(sql_query)

# 打印结果
for table in tables:
    print(table)

table1
table2