Python SQLiteでSELECT結果の件数の取得方法

Question

SELECT結果の件数を取得したいのですが、print関数で件数を取得すると、2回目の件数が0件になってしまいます。
fetchallの仕様かなと思うのですが、この場合どうコーディングすればよいか教えて頂けないでしょうか。
よろしくお願い致します。

sql = "SELECT * FROM tmp"
cur = conn.cursor()
cur.execute(sql)

print(len(cur.fetchall())) # 件数を取得できる
if len(cur.fetchall()) == 0:
    print(len(cur.fetchall())) # 0件になる
    tkm.showwarning('警告','0件のため取込できません。')

Answer 1

おそらくですが、同じ条件で2回fetchallを行っているからだと思います。
一度目のfetchallで、その条件に当てはまる全てのデータがフェッチされ、二度目のフェッチでは、フェッチするものが残っていないためそのようになっているのだと思います。

その場合は、
fetchallで取得できるデータを一度変数で受け取って繰り返し利用した方がいいかと思います。

または

cur.close()
cur = conn.cursor()

としてカーソルを一旦クローズすることでリセットさせてもできるかと思います。

Answer 2

fetchall すると残りのクエリ結果をすべて取得するので、fetchall を実行した直後に再度実行すると何も値は返りません。

まず件数を取得したい場合は select count(*) from tmp のようにsqliteの count 関数を使った方が効率がよいです。この場合は fetchone で返される値がそのまま件数になります。クエリ結果は１つしかないので、fetchall を使う必要はありません。

それで、もう一度取得したい場合は同じクエリでもう一回 execute すればいいです。

Comments

[WMS]

ご回答ありがとうございます。
質問を投稿した後、このやり方も考えたのですが、SQLを2回実行することに違和感を感じており、ほかに方法おがないか調査しておりました。
ご教授頂いた方法は、ケースバイケースでデータ件数だけを返すような共通関数を作成する手もありですね、参考にさせて頂きます。