是否有“with conn.cursor() as...”的方式来使用 Sqlite？

Question

而不是使用：

import sqlite3
conn = sqlite3.connect(':memory:')
c = conn.cursor()
c.execute(...)
c.close()

是否可以使用 Pythonic 成语：

with conn.cursor() as c:
    c.execute(...)

它似乎不起作用：

AttributeError: __exit__

注：关闭，因为游标是很重要的这个。

Answer 1

您可以使用contextlib.closing：

import sqlite3
from contextlib import closing

conn = sqlite3.connect(':memory:')

with closing(conn.cursor()) as cursor:
    cursor.execute(...)

这是有效的，因为在 with 块之后closing(object)自动调用close()传入对象的方法。

Answer 2

一个更简单的替代方案是将连接对象与上下文管理器一起使用，如文档中指定的。

with con:
    con.execute(...)

如果您坚持使用光标（因为原因），那么为什么不创建自己的包装类呢？

class SafeCursor:
    def __init__(self, connection):
        self.con = connection

    def __enter__(self):
        self.cursor = self.con.cursor()
        return self.cursor

    def __exit__(self, typ, value, traceback):
        self.cursor.close()

然后你将这样调用你的类：

with SafeCursor(conn) as c:
    c.execute(...)