fetchmany で fetch する

時々使うので自分のメモ的に。

定義

def fetchmany(cursor, size=1000):
    """
    1000件づつ fetch する
    """
    while True:
        rows = cursor.fetchmany(size)
        if not rows:
            break
        for row in rows:
            yield row

利用

# cursor を取得
import MySQLdb
conn = MySQLdb.connect()
conn.cursorclass = MySQLdb.cursors.DictCursor  # select した結果を dict で取りたい時等
cursor = conn.cursor()

# 検索
cursor.execute('SELECT foo FROM bar WHERE baz = %s', ('piyo', ))
for row in fetchmany(cursor):
    print(row['foo'])

参考

• 元ネタ: Use generators for fetching large db record sets (Python recipe)
• PEP 249

僕が所属している会社が社員募集中みたいです。なんか、 Python 書いてみてーなと思った人は応募してみてくださいな。 (すてま)