1.用 functools.wraps 定义函数修饰器
-
装饰器可以对函数进行封装,但是会改变函数信息
-
使用 functools 的 warps 可以解决这个问题
def trace(func):
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
print('%s(%r, %r) -> %r' % (func.__name__, args, kwargs, result))
return result
return wrapper
@trace
# 装饰器含义就是 fibonacci = trace(fibonacci)
def fibonacci(n):
"""返回第 n 个斐波那契数字。"""
if n in (0, 1):
return n
return (fibonacci(n-2) + fibonacci(n-1))
fibonacci(3)
>>>
fibonacci((1,), {}) -> 1
fibonacci((0,), {}) -> 0
fibonacci((1,), {}) -> 1
fibonacci((2,), {}) -> 1
fibonacci((3,), {}) -> 2
print(fibonacci)
>>>
<function trace.<locals>.wrapper at 0x0000021B0C068700>
trace 函数所返回的值,是它内部定义的那个 wrapper。而我们又用 trace 来修饰原有的 fibonacci 函数,于是,Python 就会把修饰器内部的那个 wrapper 函数,赋值给当前模块中与原函数同名的 fibonacci 变量。对于调试器和对象序列化器等需要使用内省机制的那些工具来说,这样的行为会干扰它们的正常运作。会使help函数失效:help(fibonacci)。
warps 本身也是修饰器,它可以帮助开发者编写其他修饰器。将 warps 修饰器运用到 wrapper 函数之后,它就会把与内部函数相关的元数据全部复制到外围函数。
from functools import *
def trace(func):
@wraps(func)
def wrapper(*args, **kwargs):
result = func(*args, **kwargs)
print('%s(%r, %r) -> %r' % (func.__name__, args, kwargs, result))
return result
return wrapper
@trace
# 装饰器含义就是 fibonacci = trace(fibonacci)
def fibonacci(n):
"""返回第 n 个斐波那契数字。"""
if n in (0, 1):
return n
return (fibonacci(n-2) + fibonacci(n-1))
print(fibonacci)
>>>
<function fibonacci at 0x0000026E0A9DE3A0>
help(fibonacci)
>>>
Help on function fibonacci in module __main__:
fibonacci(n)
返回第 n 个斐波那契数字。
2.考虑以 contextlib 和 with 语句来改写可复用的 try/finally 代码