Python装饰器实现数据库连接的封装实践

装饰器是Python中的一种高级特性，用于实现在不修改原有代码的情况下，对函数进行功能扩展。在数据库连接的封装中，装饰器可以用于自动创建和关闭数据库连接，提高代码的复用性和可维护性。

要实现数据库连接的封装，首先需要引入相关的库，比如pymysql库用于连接MySQL数据库。然后，可以定义一个装饰器函数，用于自动创建和关闭数据库连接。

下面是一个示例代码，演示了如何使用装饰器实现数据库连接的封装：

import pymysql
from functools import wraps

def db_connection(func):
    @wraps(func)  # 保留原函数的元数据
    def wrapper(*args, **kwargs):
        # 创建数据库连接
        conn = pymysql.connect(host='localhost', user='root', password='password', db='database')
        try:
            # 调用原函数并传入数据库连接对象
            result = func(conn, *args, **kwargs)
        finally:
            # 关闭数据库连接
            conn.close()
        # 返回原函数的返回值
        return result
    return wrapper

@db_connection
def fetch_data(conn, query):
    cursor = conn.cursor()
    cursor.execute(query)
    data = cursor.fetchall()
    return data

query = 'SELECT * FROM table'
data = fetch_data(query)
print(data)

在上面的代码中，我们定义了一个装饰器函数db_connection，并在该装饰器函数中实现了数据库连接的逻辑。装饰器函数接受一个函数作为参数，并在内部定义了一个新函数wrapper。在wrapper函数中，我们首先创建了数据库连接对象conn，然后调用原函数func并传入conn对象和原函数的参数。最后，我们在finally块中关闭数据库连接，并返回原函数的返回值。

通过在需要连接数据库的函数上添加@db_connection装饰器，我们可以实现自动创建和关闭数据库连接的功能。在上述示例代码中，我们定义了一个函数fetch_data，用于执行查询语句并返回查询结果。使用装饰器后，我们在调用fetch_data函数时不需要手动创建和关闭数据库连接，装饰器会自动处理这些细节。

装饰器可以在各种场景中起到简化代码和提高代码复用性的作用。在数据库连接的封装中，装饰器可以帮助我们避免重复的数据库连接代码，提高代码的可读性和可维护性。同时，装饰器还可以轻松地应用到其他需要数据库连接的函数上，实现更多功能的封装和扩展。