Lwaxx的博客

深入探讨 Python 中的装饰器和上下文管理器

发表于 分类于 ， 阅读次数： Valine：

Python 作为一门灵活而强大的语言，提供了许多高级特性，其中装饰器（Decorators）和上下文管理器（Context Managers）是其中两个非常有用的概念。这两个功能性特性提供了对代码结构和行为进行修改和控制的强大工具。它们允许程序员在不修改源代码的情况下，添加、修改或扩展函数或类的功能，帮助编写更优雅、更干净的代码，同时提高代码的可重用性和可维护性。

装饰器（Decorators）

装饰器是函数的函数，它接受一个函数作为参数，并返回一个新的函数。它们提供了一种简洁的方式来包装或修改函数的行为。通过装饰器，可以在不改变原始函数代码的情况下，添加额外的功能或逻辑，如日志记录、性能计时、权限检查等。这种能力使得装饰器成为Python中函数式编程范式的强大工具之一。

基本语法

1
2
3
4
5
6
7
8
9
10
def decorator_function(func):
    def wrapper(*args, **kwargs):
        # 添加装饰逻辑
        return func(*args, **kwargs)
    return wrapper

@decorator_function
def some_function():
    # 函数体
    pass

举例说明：

1. 计时器装饰器

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import time

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        result = func(*args, **kwargs)
        end_time = time.time()
        print(f"Execution time: {end_time - start_time} seconds")
        return result
    return wrapper

@timer
def some_function():
    time.sleep(2)
    print("Function executed")

some_function()
# 输出：Function executed
#      Execution time: 2.000123 seconds

2. 权限检查装饰器

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
def check_permission(func):
    def wrapper(*args, **kwargs):
        if user_has_permission():
            return func(*args, **kwargs)
        else:
            raise PermissionError("Permission denied")
    return wrapper

@check_permission
def sensitive_operation():
    print("Operation executed")

sensitive_operation()
# 如果用户有权限，输出：Operation executed
# 如果用户无权限，抛出 PermissionError

上下文管理器（Context Managers）

上下文管理器提供了对资源进行安全获取和释放的机制，即使在出现异常时也能确保资源的释放。这对于处理文件、数据库连接或其他需要资源管理的情况特别有用。上下文管理器可以使用 with 语句来确保在代码块执行前获取资源，在代码块执行后释放资源，保证资源的正确处理。

基本语法

1
2
3
4
5
6
7
8
9
10
11
12
class CustomContextManager:
    def __enter__(self):
        # 分配资源
        return resource
    
    def __exit__(self, exc_type, exc_value, traceback):
        # 释放资源
        pass

with CustomContextManager() as resource:
    # 使用资源的代码块
    pass

举例说明：

3. 文件操作的上下文管理器

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class FileManager:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
    
    def __enter__(self):
        self.file = open(self.filename, self.mode)
        return self.file
    
    def __exit__(self, exc_type, exc_value, traceback):
        self.file.close()

with FileManager("example.txt", "w") as file:
    file.write("Hello, Context Manager!")
# 文件 example.txt 被正确地写入数据，并在代码块结束时自动关闭

4. 数据库连接的上下文管理器

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import sqlite3

class DatabaseConnection:
    def __init__(self, database):
        self.database = database
    
    def __enter__(self):
        self.connection = sqlite3.connect(self.database)
        return self.connection
    
    def __exit__(self, exc_type, exc_value, traceback):
        self.connection.close()

with DatabaseConnection("example.db") as conn:
    cursor = conn.cursor()
    cursor.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)")
# 在代码块中成功创建数据库连接，并在结束时自动关闭连接

高阶概念与应用

装饰器链：多个装饰器可以被串联使用，以添加多个功能。
上下文管理器的异步支持：async with 语句在异步代码中管理异步资源的获取和释放。
这些高级特性提供了对 Python 代码逻辑和资源管理更细粒度的控制，使得代码更具灵活性和可维护性。

结语

装饰器和上下文管理器是 Python 中强大而灵活的特性，它们可以使代码更简洁、更易于维护，并且提供了许多便利。