如何在Python中超时后取消SQL查询？

针对你遇到的查询无限执行的问题，以下是几种在pyodbc中实现10分钟超时自动取消的可行方案：

1. 连接字符串全局设置超时 #

pyodbc支持在连接字符串中通过QueryTimeout参数设置全局查询超时，单位为秒（10分钟=600秒）。所有通过该连接执行的查询都会继承这个超时时间，实现起来最简单直接：

import pyodbc

# 构建带超时的连接字符串
connection_string = (
    "DRIVER={ODBC Driver 17 for SQL Server};"
    "SERVER=你的服务器地址;"
    "DATABASE=目标数据库;"
    "UID=用户名;"
    "PWD=密码;"
    "QueryTimeout=600;"  # 10分钟超时设置
)

try:
    cnxn = pyodbc.connect(connection_string)
    cursor = cnxn.cursor()
    cursor.execute(你的查询语句)
    cnxn.commit()
except pyodbc.Error as e:
    print(f"查询超时或执行失败: {e}")
finally:
    if 'cnxn' in locals():
        cnxn.close()

2. 针对单个Cursor设置超时 #

如果需要给不同查询设置不同超时时间，可以直接修改Cursor的timeout属性，优先级高于连接字符串的全局设置：

import pyodbc

cnxn = pyodbc.connect(你的基础连接字符串)
cursor = cnxn.cursor()
cursor.timeout = 600  # 给当前cursor的查询设置10分钟超时

try:
    cursor.execute(你的查询语句)
    cnxn.commit()
except pyodbc.Error as e:
    print(f"查询超时或执行失败: {e}")
finally:
    cursor.close()
    cnxn.close()

3. 多线程+强制关闭连接（兼容特殊场景） #

如果上述两种方法因驱动或查询类型（比如复杂存储过程）不生效，可以用多线程执行查询，超时后通过关闭连接中断执行：

import pyodbc
import threading

def run_query(conn_str, query, result_container, error_container):
    cnxn = None
    try:
        cnxn = pyodbc.connect(conn_str)
        # 将连接存入共享容器，方便超时后关闭
        result_container["conn"] = cnxn
        cursor = cnxn.cursor()
        cursor.execute(query)
        # 如果需要获取查询结果，可存入容器
        result_container["data"] = cursor.fetchall()
        cnxn.commit()
    except Exception as e:
        error_container["msg"] = str(e)
    finally:
        if cnxn:
            cnxn.close()

# 初始化共享容器存储结果、连接和错误信息
result = {}
error = {}

# 启动线程执行查询
query_thread = threading.Thread(
    target=run_query,
    args=(你的连接字符串, 你的查询语句, result, error)
)
query_thread.start()

# 等待10分钟，超时则中断
query_thread.join(600)

if query_thread.is_alive():
    print("查询超时，正在终止...")
    # 关闭连接中断查询
    if "conn" in result:
        result["conn"].close()
    print("查询已终止")
else:
    if error:
        print(f"查询执行失败: {error['msg']}")
    else:
        print("查询执行完成")
        # 可处理result["data"]中的查询结果

关键注意事项 #

• 驱动版本：确保使用ODBC Driver 17及以上版本的SQL Server驱动，老版本可能不支持QueryTimeout参数。
• 异常处理：超时触发时会抛出pyodbc.Error，必须用try-except捕获，避免程序崩溃。
• 存储过程场景：部分复杂存储过程可能无法被连接级超时中断，此时多线程+关闭连接的方式更可靠。

内容的提问来源于stack exchange，提问作者principal-ideal-domain