python SQLITEのcursorは前に戻せないのですか

Question

タイトル通りの疑問です。

シンプルなことなのですが、

c=conn.cursor
c.execute(SQL)

for row in c:
   処理A

for row in c:
   処理B

このように書くと、Aの時点でカーソルが最後の行に移動してしまい、
Bの処理はできなくなってしまうと思います。

ここで、カーソルを前に戻すというメソッドなどは存在しないのでしょうか。
ドキュメントも一通り眺めて見たつもりなのですがそれらしきものが見当たりませんでした。

また、もし戻せないとすると、再度
c.execute(SQL)
を実行することになるのでしょうか（それも無駄が多いような気がして）。

ご指導よろしくお願いいたします。

Answer 1

ここで、カーソルを前に戻すというメソッドなどは存在しないのでしょうか。

はい、PythonのSQLiteはCursor#scrollを実装していないので、存在しません。
pysqliteのcursor.c

fetchone/fetchmany/fetchallがlistを返すので、fetch関数を使う手もあります。

以下はサンプルコードです。

# -*- coding: utf8 -*-
import sqlite3
from contextlib import closing

tablename = 'sample'


def insert_data(cur: sqlite3.Cursor) -> None:
    import random
    random.seed(42)
    row_count = 0
    for i in range(10):
        sql = f"insert into {tablename} values (?, ?)"
        params = (i, random.randint(0, 500))
        cur.execute(sql, params)
        row_count += cur.rowcount

    print(f"rowcount:{row_count}")


def select_data(cur: sqlite3.Cursor) -> None:
    sql = f"SELECT * FROM {tablename} ORDER BY id"

    cur.execute(sql)
    rows = cur.fetchall()
    for row in rows:
        print(row)
    print("#" * 40)
    for row in rows:
        print(row)


def main() -> None:
    db_name = 'sample.db'
    with closing(sqlite3.connect(db_name)) as conn:
        conn.set_trace_callback(print)
        cur = conn.cursor()
        cur.execute(f"drop table if exists {tablename}")
        cur.execute(f"create table {tablename} (id, rnd)")
        insert_data(cur)
        select_data(cur)
        # コミット忘れに注意！
        conn.commit()


if __name__ == '__main__':
    main()

Comments

[退会済みユーザー]

ありがとうございます。やはり戻せないのですね。ちなみにfetchallと再度executeではパフォーマンスの違いはあるでしょうか。自分で少し試したところ、fetchallがやたらに重かったことがあったので、、。よければお教えください。

[umyu]

>qwdcf_mag
1,executeはジェネレータを返し、fetchallは内部で結果のlistを生成するので、その文メモリを消費します。
2,パフォーマンスが要求される部分に関しては、きちんとプロファイル(cprofile&snakeviz)を取って対応したほうがいいとおもいますー。。
3,PythonのSQLiteが該当するかは分かりませんが、経験則で回答するとSQLiteで重いと仰ってる人に共通するのはSQLiteの暗黙のトランザクションについて知らない人が多いという点ですね。

Answer 2

直接の回答ではないですが。

単純な例で書かれていて、実際にはどのようなことをやりたいのかはわかりませんが、

for row in c:
  処理A
  処理B

じゃダメなんですか？同じデータ（群）に対して処理するんですよね？

Comments

[退会済みユーザー]

ありがとうございます。全く別系統の処理をして２つの出力を出すので、分けたいなという状況です。