装饰器设置功能的属性

Question

我希望仅当登录用户具有所需的权限级别时，其他功能才可执行。

为了使我的生活更加复杂，我想使用装饰器。下面，我尝试permission在“装饰”功能上设置属性-如下所示。

def permission(permission_required):
    def wrapper(func):
        def inner(*args, **kwargs):
            setattr(func, 'permission_required', permission_required)
            return func(*args, **kwargs)
        return inner
    return wrapper

@permission('user')
def do_x(arg1, arg2):

    ...

@permission('admin')
def do_y(arg1, arg2):
    ...

但是当我这样做时：

fn = do_x
if logged_in_user.access_level == fn.permission_required:
    ...

我得到一个错误 'function' object has no attribute 'permission_required'

我想念什么？

Answer 1

您正在检查内部（包装）函数上的属性，但在原始（包装）函数上进行了设置。但是，您 根本 需要包装函数：

def permission(permission_required):
    def decorator(func):
        func.permission_required = permission_required
        return func
    return decorator

你的装饰需要返回 的东西 那将取代原有的功能。原始函数本身（添加了属性）可以很好地完成此任务，因为您要做的只是向其添加一个属性。

如果仍然需要包装器，则在 包装器函数 上设置属性：

from functools import wraps

def permission(permission_required):
    def decorator(func):
        @wraps(func)
        def wrapper(*args, **kwargs):
            # only use a wrapper if you need extra code to be run here
            return func(*args, **kwargs)
        wrapper.permission_required = permission_required
        return wrapper
    return decorator

毕竟，您要用装饰器返回的包装器替换包装的函数，因此这就是您要在其上查找属性的对象。

我还将@functools.wraps()装饰器添加到包装器中，该装饰器将重要的标识信息和其他有用的东西复制func到包装器中，从而使使用起来更加容易。