如何使用pymssql在Python中调用返回多结果集的存储过程

嘿，我刚好处理过类似的需求，用pymssql获取存储过程返回的多个结果集其实并不复杂，和你熟悉的Java Spring JDBC的思路异曲同工——咱们可以用类似字典（对应Java里的Map）的方式来按“键”访问每个结果集，具体操作如下：

核心思路 #

pymssql的游标对象提供了nextset()方法，用来切换到下一个结果集。我们可以先执行存储过程，然后依次获取每个结果集，把它们存入字典并自定义键名，之后就可以像Java里那样通过键来访问对应的结果集了。

完整示例代码 #

import pymssql

# 1. 建立数据库连接
conn = pymssql.connect(
    server="你的数据库服务器地址",
    user="用户名",
    password="密码",
    database="目标数据库名"
)

try:
    # 2. 创建游标对象
    cursor = conn.cursor()
    
    # 3. 调用存储过程（如果有参数就按格式传入）
    # 示例：带两个参数的存储过程
    cursor.execute("EXEC YourMultiResultProc @Param1 = %s, @Param2 = %s", ("参数值1", "参数值2"))
    
    # 4. 用字典存储结果集，自定义键名对应每个结果集
    result_map = {}
    
    # 获取第一个结果集，比如命名为'user_list'
    result_map['user_list'] = cursor.fetchall()
    
    # 切换到下一个结果集，存在的话继续获取
    if cursor.nextset():
        result_map['order_summary'] = cursor.fetchall()
        
        # 如果还有更多结果集，继续切换并存储
        if cursor.nextset():
            result_map['product_stats'] = cursor.fetchall()
    
    # 5. 现在就可以像Java那样通过键访问结果集了
    print("用户列表：", result_map['user_list'])
    print("订单汇总：", result_map['order_summary'])

finally:
    # 6. 务必关闭游标和连接，避免资源泄漏
    cursor.close()
    conn.close()

处理不确定数量的结果集 #

如果不确定存储过程会返回多少个结果集，用循环来处理会更灵活：

result_map = {}
set_count = 1

while True:
    # 获取当前结果集
    current_set = cursor.fetchall()
    result_map[f'result_set_{set_count}'] = current_set
    
    # 尝试切换到下一个结果集，没有的话退出循环
    if not cursor.nextset():
        break
    set_count += 1

# 遍历所有结果集
for key, data in result_map.items():
    print(f"{key} 数据：", data)

注意事项 #

• 如果结果集数据量很大，不要用fetchall()一次性获取所有数据，改用fetchmany(size=1000)分批读取，避免内存溢出。
• 如果存储过程包含输出参数，可以通过cursor.execute()执行后，用cursor.fetchone()或者直接访问游标属性来获取（具体取决于参数定义）。
• 操作完成后一定要关闭游标和连接，最好用try...finally块确保资源释放。

内容的提问来源于stack exchange，提问作者RAFJR