装饰器会用一个动态创建的新函数替换原来的,新函数缺少很多原函数的属性,如 docstring和名字

python内置的 functools 模块通过其 update_wrapper 函数解决这个问题

手工调用 update_wrapper 创建装饰器很不方便,所以 functools提供了名为 wraps 的装饰器

inspect 从装饰器函数中获取相应的参数,通过 getcallargs 函数把参数转化为 dict格式

下面使用示例:

# -*- coding:utf-8 -*-
# @author: pemakoa@gmail.com

import functools
import inspect


def check_is_admin(f):
    @functools.wraps(f)
    def wrapper(*args, **kwargs):
        func_args = inspect.getcallargs(f, *args, **kwargs)
        if func_args['username'] != 'admin':
            raise Exception("This user is not allowed to get food")
        return f(*args, **kwargs)
    return wrapper


@check_is_admin
def get_food(username, type="chocolate"):
    return type + " nom nom nom!"

继续阅读关于 的文章



Fork me on GitHub