python - 我应该为每个 Sqlite3 事务调用 connect() 和 close() 吗？

Question

我想编写一个 Python 模块来为我的应用程序抽象出数据库事务。我的问题是我是否需要为每笔交易调用 connect() 和 close()？在代码中:

import sqlite3

# Can I put connect() here?
conn = sqlite3.connect('db.py')

def insert(args):
    # Or should I put it here?
    conn = sqlite3.connect('db.py')
    # Perform the transaction.
    c = conn.cursor()
    c.execute(''' insert args ''')
    conn.commit()
    # Do I close the connection here?
    conn.close()

# Or can I close the connection whenever the application restarts (ideally, very rarely)
conn.close()

我在数据库方面经验不多，所以如果您能解释为什么一种方法优于另一种方法，我将不胜感激。

Answer 1

您可以重复使用同一个连接。您还可以将连接(和游标)用作上下文管理器，这样您就不需要显式调用 close。

def insert(conn, args):
    with conn.cursor() as c:
        c.execute(...)
    conn.commit()

with connect('db.py') as conn:
    insert(conn, ...)
    insert(conn, ...)
    insert(conn, ...)

没有理由关闭与数据库的连接，而且每次重新打开连接的代价都很高。 (例如，您可能需要建立 TCP session 以连接到远程数据库。)