pythonのpsycopg2においてfor文を用いたcursorの使い方としてどちらがよいか？

Question

問題点

pythonでデータベースへの接続し、for文でsqlを実行した際にcursorの使い方としてどちらが良いか分かりませんでした。
毎回cursorを開いて閉じた方がいいのか、cursorを上書きしていく方がいいのか、分からないため、よろしければアドバイスを頂けないでしょうか。
コード例を下に示します。

conn=psycopg2.connect(connect)
def execute_1(sql):
    with conn:
        with conn.cursor() as cur:
            cur.execute(sql)

def execute_2(sql):
    with conn:
       cur = conn.cursor()
        cur.execute(sql)

for i in range(1000000):
    sql=f"update table set a = {i};"
    execute_1(sql)
    execute_2(sql)

conn.close()

アドバイスを頂けると嬉しいです

Answer 1

どちらがいいかというのはどういう基準での話なのかわかりませんが、まず 2 つ目の execute_2() 関数は、cur.close() する必要があると思います。そういう意味では、execute_1() の方がいいと思います。

それとは別に、そもそも for 文をコンテキストマネージャーの中で回した方がいい気がします。

import psycopg2

DSN: str = "path/to/database"
SQL: str = "update table set a = {};"

conn = psycopg2.connect(DSN)


def execute(sql: str):
    try:
        with conn:
            with conn.cursor() as cur:
                for i in range(1000000):
                    cur.execute(sql.format(i))
    finally:
        conn.close()


execute(SQL)

個人的には、conn.close() するのも面倒なので、以下のようにするのが好きです。

import psycopg2

DSN: str = "path/to/database"
SQL: str = "update table set a = {};"


def execute(sql: str) -> None:
    with psycopg2.connect(dsn=DSN) as conn:
        with conn.cursor() as curs:
            for i in range(1000000):
                curs.execute(sql.format(i))


execute(SQL)

Comments

[alfald]

アドバイスありがとうございます。
コードを簡略化しすぎたため、やりたいことを上手く伝えることができませんでした。
```https://github.com/mushahiroyuki/python-pro/blob/master/ch06/bark1/database.py```
上記のサイトの5~17行を参考にコードを作成したため、あのようなコードになってしまいました。
上記のサイトでは、cursorをcloseして無いように感じたため、今回の質問を上げさせてもらいました。

[Demerara]

参照先のコードは、sqlite3 を使用しています。sqlite3 ならば cursor を close する必要はありません。

公式ドキュメントを参照してください。
sqlite3: <https://docs.python.org/3/library/sqlite3.html>
psycopg2: <https://www.psycopg.org/docs/usage.html>

[alfald]

そうだったんですね。
公式ドキュメントの例のコードにはcur.close()がなかったけどconn.closeしてるから省略していて、cursorオブジェクトの中にclose()があったため必要だと思ってました。
お答えいただきありがとうございます。