如何正确关闭 SQLite 数据库连接-Python教程-PHP中文网

如何正确关闭 SQLite 数据库连接

霞舞

发布： 2025-10-17 14:59:11

原创

851人浏览过

如何正确关闭 sqlite 数据库连接

本文介绍了在 Python 中使用 SQLite 数据库时，如何通过上下文管理器和装饰器来安全、有效地管理数据库连接的打开和关闭。重点讲解了 __enter__ 和 __exit__ 方法的正确使用，以及如何在 __exit__ 方法中处理异常和提交事务，并提供了一个完整的示例代码，展示了如何将这些技术应用到实际的数据库操作中。

在使用 SQLite 数据库时，正确地管理数据库连接至关重要。忘记关闭连接可能导致资源泄漏，甚至数据损坏。Python 提供了上下文管理器和装饰器等工具，可以帮助我们更优雅、更安全地管理数据库连接。

使用上下文管理器 (with 语句)

上下文管理器允许我们在进入和退出代码块时自动执行一些操作。对于数据库连接来说，进入代码块时打开连接，退出代码块时关闭连接，非常方便。

要实现上下文管理器，需要定义一个类，并实现 __enter__ 和 __exit__ 方法。

__enter__(self): 在进入 with 语句块时调用，返回一个对象，该对象会被 as 后的变量引用。
__exit__(self, exc_type, exc_val, exc_tb): 在退出 with 语句块时调用。如果代码块中发生异常，exc_type、exc_val 和 exc_tb 分别是异常类型、异常值和 traceback 对象；如果没有异常发生，这三个参数都为 None。

以下是一个 DBConnection 类的示例，它实现了上下文管理器：

import sqlite3

class DBConnection:
    def __enter__(self):
        self.conn = sqlite3.connect('main.db')
        self.cursor = self.conn.cursor()
        return self.cursor

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.cursor:
            self.cursor.close()
        if self.conn:
            if exc_type is None:
                self.conn.commit()  # this is to commit if no exception
            self.conn.close()
        return False  # 重新抛出异常

登录后复制

注意：

__exit__ 方法接收三个参数 exc_type, exc_val, exc_tb，用于处理可能发生的异常。
在 __exit__ 方法中，应该先关闭游标 (cursor)，再关闭连接 (conn)。
在没有异常发生时，应该提交事务 (conn.commit())，确保数据被保存。
return False 表示如果发生异常，则重新抛出异常，让调用者处理。如果返回 True，则会阻止异常传播。

使用装饰器

装饰器可以用来包装函数，在函数执行前后执行一些操作。我们可以使用装饰器来自动打开和关闭数据库连接。

来画数字人直播

来画数字人自动化直播，无需请真人主播，即可实现24小时直播，无缝衔接各大直播平台。

查看详情

以下是一个 db_connection 装饰器的示例：

def db_connection(func):
    def wrapper(*args, **kwargs):
        with DBConnection() as cursor:
            return func(cursor, *args, **kwargs)
    return wrapper

登录后复制

这个装饰器接受一个函数作为参数，并返回一个新的函数 wrapper。wrapper 函数使用 DBConnection 上下文管理器来获取游标，然后调用原始函数，并将游标作为参数传递给它。

示例

以下是一个完整的示例，展示了如何使用上下文管理器和装饰器来管理数据库连接：

import sqlite3

class DBConnection:
    def __enter__(self):
        self.conn = sqlite3.connect('main.db')
        self.cursor = self.conn.cursor()
        return self.cursor

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.cursor:
            self.cursor.close()
        if self.conn:
            if exc_type is None:
                self.conn.commit()  # this is to commit if no exception
            self.conn.close()
        return False

def db_connection(func):
    def wrapper(*args, **kwargs):
        with DBConnection() as cursor:
            return func(cursor, *args, **kwargs)
    return wrapper

class Database:
    @db_connection
    def some_query(self, cursor, *args):
        query = "SELECT * FROM your_table"  # 替换为你的 SQL 查询
        cursor.execute(query)
        results = cursor.fetchall()
        return results

登录后复制

使用方法：

db = Database()
results = db.some_query()
print(results)

登录后复制

在这个例子中，some_query 方法使用了 @db_connection 装饰器，这意味着每次调用 some_query 方法时，都会自动打开数据库连接，执行查询，并在方法执行完毕后自动关闭连接。

总结

通过使用上下文管理器和装饰器，我们可以更方便、更安全地管理 SQLite 数据库连接。这可以避免资源泄漏，并提高代码的可读性和可维护性。记住在 __exit__ 方法中处理异常和提交事务，以确保数据的完整性。

以上就是如何正确关闭 SQLite 数据库连接的详细内容，更多请关注php中文网其它相关文章！

大家都在看：

python定义数值类型变量的方法 python决策树算法是什么 Python变量怎么定义_Python变量定义方法与注意事项 Python集合怎么定义_Python集合的定义与使用场景分析 Python装饰器怎么理解_Python装饰器概念与实际应用方法