pyodbc 使用 DB2 从存储过程返回多个游标

Question

我有一个 python 程序，它从 db2 数据库调用存储过程。我正在使用 results = curve.fetchall() 来处理存储过程的结果。但是，我的存储过程返回两个游标。 results 仅包含第一个。我需要一种方法来循环遍历尽可能多的光标。我希望 fetchmany() 是我的答案，但事实并非如此。

我需要能够执行多个结果集，因为我正在编写的程序只能调用一个存储过程。要回去并使其能够呼叫两个需要花费很多时间。除了这些事情之一之外，我还需要让 10 个游标返回。一切都是动态的，所以应用程序不知道它正在运行什么程序，它只是获取数据并将其吐入excel而不知道其含义。我需要一个游标用于数据，另一个游标用于不同类型的计数和总计。

我正在寻找一个内置函数来执行此操作，或者甚至可能是一个不同的库，因为我已经完成了谷歌搜索，看起来 pyodbc 并没有为 DB2 执行此操作。 DB2 是一个要求。

Answer 1

使用光标的 nextset() 方法：https:// github.com/mkleehammer/pyodbc/wiki/Cursor#nextset

示例代码：

# fetch rows from first set
rows = cursor.fetchall()    
# process first set rows here

# advance to next result set
while (cursor.nextset()):    
    # fetch rows from next set, discarding first
    rows = cursor.fetchall()    
    # process next set rows here

如果有其他结果集可用，nextset() 将返回 True，并且后续光标fetch 方法将从下一组返回行。如果没有其他可用集，该方法将返回 None。

Answer 2

只是为了记录而做一个小小的简化：

while True:    
    rows = cursor.fetchall()
    # process all result sets in the same place
    if not cursor.nextset():
        break