使用Python脚本查找SQL查询中的所有表

import re
import pymysql

conn = pymysql.connect(host='数据库主机地址', user='用户名', password='密码', database='数据库名')

cursor = conn.cursor()

def find_tables(sql_query):
    # 使用正则表达式匹配SQL查询中的表名
    table_names = re.findall(r'FROM\s+([\w_]+)', sql_query, re.IGNORECASE)
    table_names += re.findall(r'JOIN\s+([\w_]+)', sql_query, re.IGNORECASE)
    table_names += re.findall(r'UPDATE\s+([\w_]+)', sql_query, re.IGNORECASE)
    table_names += re.findall(r'INTO\s+([\w_]+)', sql_query, re.IGNORECASE)
    table_names += re.findall(r'TABLE\s+([\w_]+)', sql_query, re.IGNORECASE)
    table_names += re.findall(r'ALTER\s+TABLE\s+([\w_]+)', sql_query, re.IGNORECASE)
    
    # 去除重复的表名
    table_names = list(set(table_names))
    
    return table_names

sql_query = 'SELECT * FROM users JOIN orders ON users.id = orders.user_id'
tables = find_tables(sql_query)
print(tables)