python 之内置模块

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 代码

上一篇:python *args和**kwargs


下一篇:Django(27)Django类视图添加装饰器