Python装饰器进阶技巧：参数检查与返回值处理

"Python装饰器使用你可能不知道的几种姿势" Python装饰器是编程中的一个重要工具，它们允许我们修改或增强函数、方法或类的行为，而无需改变其原始代码。装饰器在Python中广泛用于日志记录、性能测试、权限控制等场景。下面将详细介绍两种在装饰器中操作被装饰函数的技巧。 ### 一、在装饰器中获取被装饰函数的参数 在装饰器中获取被装饰函数的参数通常是为了在函数执行前后添加额外的操作。例如，上述描述中的装饰器`mydecorator`用于检查web接口的访问来源。装饰器内部定义了一个内部函数`wrapped`，这个内部函数接收`*args`和`kwargs`作为参数，以便捕获被装饰函数的所有位置和关键字参数。 ```python def mydecorator(func): def wrapped(*args, kwargs): # 在这里可以对传入的参数进行操作 if args[0]['header'] == 'spider': print("code: 400") # 如果访问来源是爬虫，返回错误 else: result = func(*args, kwargs) # 调用原函数并返回结果 return result return wrapped ``` 使用`@mydecorator`装饰`request_page`函数，装饰器会自动插入到函数调用之前，检查`request`参数中的`header`字段。这种装饰器的使用方式与直接在`request_page`函数中调用`mydecorator`函数等效，但装饰器提供了更好的代码复用和可读性。 ### 二、在装饰器中获取函数的返回值 除了操作参数，装饰器还可以用来处理函数的返回值。这在你需要在函数执行后添加额外逻辑，比如缓存结果、验证返回数据或者转换格式时非常有用。以下是一个简单的例子： ```python def log_return_value(func): def wrapper(*args, kwargs): result = func(*args, kwargs) print(f"函数{func.__name__}返回了：{result}") return result return wrapper @log_return_value def calculate_area(radius): return 3.14 * radius 2 calculate_area(5) # 打印出计算的面积并返回 ``` 在这个例子中，`log_return_value`装饰器记录了`calculate_area`函数的返回值，并将其打印出来，同时保持原始的返回值不变。 ### 装饰器的其他用法和注意事项 1. 装饰器链：你可以连续使用多个装饰器，每个装饰器都会按顺序依次执行。 2. 带参数的装饰器：通过接受参数的装饰器，可以增加装饰器的灵活性，使其能适应不同情况。 3. 元类和类装饰器：Python的元类也可以用来装饰类，改变类的行为。 4. 使用`functools.wraps`：为了保留被装饰函数的元信息（如名称、文档字符串等），应使用`functools.wraps`装饰器。 装饰器是Python中强大的特性之一，熟练掌握装饰器的使用可以提升代码的组织和维护性，让代码更加模块化和易于理解。在实际开发中，根据需求选择合适的装饰器实现，可以极大地提高代码的效率和质量。