关于mysql游标和迭代器

Question

想象一下,我有一个mysql游标和数据读取.数据量可能非常大,我想每次处理一行.

一个简单直接的方式可能是这样的:

while True:
    row = cursor.fetchone()
    if not row: break
    .....

但这看起来不太好,所以我想知道这种方式是否像想象的那样有效:

for row in iter(cursor.fetchall())

我想知道的是:如果我使用这种iter(cursor.fetchall())方式,它是先获取所有数据还是一次只取一行？

任何想法都表示赞赏.

谢谢

Answer 1

将MySQLdb光标类实现了迭代器协议,所以你可以简单地这样做:

cursor.execute(sql)
for row in cursor:
    print row
    ...

来自MySQLdb.cursors.BaseCursor的相关代码:

def __iter__(self):
    return iter(self.fetchone, None)