如何在类中关闭SQLite 3数据库连接？当前实现是否最优正确？

先直接说你当前代码存在的几个关键问题：

• __exit__方法参数缺失：Python上下文管理器的__exit__必须接收exc_type, exc_val, exc_tb三个参数，你现在的写法会导致运行时抛出参数不匹配的错误。
• 装饰器缩进错误：return wrapper被缩进在with代码块里了，这会导致装饰器返回的是None，而非包装后的函数，你的some_query方法根本无法正常执行。
• 缺失事务处理：如果涉及写入操作（增、删、改），当前代码没有提交事务，所有修改都不会被持久化到数据库；而且遇到异常时也没有回滚逻辑，容易造成数据不一致。
• 冗余的Cursor关闭：SQLite连接关闭时，关联的Cursor会自动被关闭，手动关闭Cursor属于多余操作。

下面是修正并优化后的实现：

修正后的DBConnection上下文管理器 #

import sqlite3

class DBConnection:
    def __enter__(self):
        self.conn = sqlite3.connect('main.db')
        # 显式开启事务（SQLite默认自动开启，这里写出来更清晰）
        self.conn.execute('BEGIN')
        self.cursor = self.conn.cursor()
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        try:
            if exc_type is None:
                # 无异常时提交事务
                self.conn.commit()
            else:
                # 发生异常时回滚，避免脏数据
                self.conn.rollback()
        finally:
            # 无论是否异常，最终都关闭连接
            self.conn.close()

修正后的装饰器 #

def db_connection(func):
    def wrapper(*args, **kwargs):
        with DBConnection() as db:
            return func(db.cursor, *args, **kwargs)
    return wrapper  # 注意缩进，必须在with代码块外面

修正后的Database类示例 #

class Database:
    @db_connection
    def some_query(self, cursor, record_id):
        # 示例查询，注意参数用占位符防止SQL注入
        query = "SELECT * FROM your_table WHERE id = ?"
        cursor.execute(query, (record_id,))
        return cursor.fetchall()

    @db_connection
    def update_record(self, cursor, record_id, new_value):
        # 示例更新操作，事务会自动处理提交/回滚
        query = "UPDATE your_table SET value = ? WHERE id = ?"
        cursor.execute(query, (new_value, record_id))

额外的最优实践提醒 #

• 你的思路是对的：用上下文管理器自动管理连接的创建与销毁，避免资源泄漏。
• SQLite连接是线程不安全的，不要在多线程环境下共享同一个连接，你当前每次调用方法都新建连接的方式，适合单线程或每次操作独立的场景。
• 如果是高频操作场景，可以考虑用连接池优化，但SQLite本身轻量，大多数业务场景下每次新建连接的性能损耗可以忽略。
• 永远用占位符编写SQL（比如示例中的?），不要拼接字符串，防止SQL注入。

内容的提问来源于stack exchange，提问作者Alex