python高手之路 笔记2装饰器专题 简明易懂

python装饰器

创建装饰器

• 装饰器本质上就是一个函数，这个函数接收其他函数作为参数，并将其以一个新的修改后的函数进行替换。
• 最简单的装饰器就是本体函数（identify function）

def identify(f):
    #A#
    return f

然后就可以像下面这样使用这个装饰器：

@identify
def foo():
    return 'bar'

它和下面的过程类似：

def foo():
    return 'bar'
foo = identify(foo)

在#A#处，可以把一批函数都需要进行的操作都放在这，再用装饰器的形式来装饰那批函数，来简化代码。

注册装饰器

_functions = {}
def register(f):
global _functions
_functions[f.name] = foo
return foo

使用##

@register
def foo():
return 'bar'

- 在这个例子中，函数被注册并存储在一个字典里，以便后续可以根据函数名字提取函数

#### 装饰器升级
- ```python
import functools
import inspect
def check_is_admin(f):
    @functools.wraps(f)
    #使用functools使返回的wrapper函数的属性与原函数一致
    def wrapper(*args, **kwargs):
    #args得到函数传参
        func_args = inspect.getcallargs(f, *args, **kwargs)
        #返回一个将参数名字和值作为键值对的字典，不论参数的传递方式
        if func_args.get('username') != 'admin':
            raise Exception("This user is not allowed to get food")
        return f(*args, **kwargs)
        #return时执行函数，返回函数返回值，有点混，注意
    return wrapper
##使用##
@check_is_admin
def get_food(username, type='chocolate'):
    return type + " nom nom nom!"

python方法类装饰器

• 在python3中不仅可以向方法传入该类的任意实例，还可以传入任何对象，只要它包含方法期望的属性
• 静态方法 @staticmethod
• 类方法@classmethod 第一个参数cls
• 抽象方法 import abc; @abc.abstractmethod

其他

• Python Decorators: A Step-By-Step Introduction