创建装饰器

Question

我正在从 HTTP GET 请求中获取一个值。我正在检查该值是否不等于无。如果是，则发送返回消息，否则正在进行进一步处理。

def api_servers():
    installation_name = request.args.get('installation_name', '')
    if installation_name == '':
        data = {"description": "Installation Name is required"}
        return HttpResponse(json.dumps(data), status=400, mimetype='application/json')

    data = { "arms": arms_list }

    return HttpResponse(json.dumps(data), status=200, mimetype='application/json')

现在，我想使用装饰器函数检查这个条件。这就是我正在尝试的。

def wrapper(func):
    def inner(): # I dont know what to pass in the arguments. Installation_name? 
                 # Maybe? But then how do I pass Installation_name

        if installation_name == '':
            data = {"description": "Installation Name is required"}
            return HttpResponse(json.dumps(data), status=400, mimetype='application/json')

        else: 
            return func()
    return inner 

如何通过装饰器实现这一点？

编辑 当我继续进行时，我发现我需要另一个装饰器来检查data 是否为None。它还检查installation_name 是否为空，然后我需要“回收”installation_name 和@ 987654328@。我又重写了你的代码。但值正在互换，即data 得到installation_name，反之亦然。你能看看吗？谢谢

def with_description(fn):

    def inner(*args, **kwargs):
    # Precheck
        installation_name = 'inst1'
        if installation_name == '':
            return 1

        # If precheck passes
        return fn(installation_name, *args, **kwargs)

    return inner


def with_description1(fn):

    def inner(*args, **kwargs):
    # Precheck
        data = 'data1'
        if data == '':
            return 1

        # If precheck passes
        return fn(data, *args, **kwargs)

    return inner

@with_description
@with_description1
def api_servers1(installation_name,data):
    print installation_name,data



api_servers1()

Answer 1

嗯，试试看：

import functools

def with_description(fn):

    @functools.wraps(fn)  # This is not strictly necessary, just pulls info (docstrings, etc) from fn and pushes into the wrapper
    def wrapper(*args, **kwargs):
        # Precheck
        installation_name = request.args.get('installation_name', '')
        if installation_name == '':
            data = {"description": "Installation Name is required"}
            return HttpResponse(json.dumps(data), status=400, mimetype='application/json')

        # If precheck passes, call the actual function
        fn(*args, **kwargs)

    return wrapper

这样你就赢了，每个装饰的函数实际上只有在预检查通过时才会被调用。

编辑

如果您想回收安装名称，只需将其传递给函数调用即可。一种可能性如下：

fn(installation_name, *args, **kwargs)

当然，被装饰的函数必须期待那个参数。