当前位置：首页 > news >正文

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

news 2025/7/12 3:54:39

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

1.装饰器（Decorators）

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

1.1. 基本语法

def decorator_function(func):def wrapper(*args, **kwargs):# 添加装饰逻辑return func(*args, **kwargs)return wrapper@decorator_function
def some_function():# 函数体pass

1. 示例一：计时器装饰器

import timedef 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 resultreturn wrapper@timer
def some_function():time.sleep(2)print("Function executed")some_function()
# 输出：Function executed
#      Execution time: 2.000123 seconds

2. 示例二：权限检查装饰器

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

2. 上下文管理器（Context Managers）

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

2.1. 基本语法

class CustomContextManager:def __enter__(self):# 分配资源return resourcedef __exit__(self, exc_type, exc_value, traceback):# 释放资源passwith CustomContextManager() as resource:# 使用资源的代码块pass

3. 示例三：文件操作的上下文管理器

class FileManager:def __init__(self, filename, mode):self.filename = filenameself.mode = modedef __enter__(self):self.file = open(self.filename, self.mode)return self.filedef __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. 示例四：数据库连接的上下文管理器

import sqlite3class DatabaseConnection:def __init__(self, database):self.database = databasedef __enter__(self):self.connection = sqlite3.connect(self.database)return self.connectiondef __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)")
# 在代码块中成功创建数据库连接，并在结束时自动关闭连接