如何使用装饰器为函数参数设置默认值?

时间:2016-04-06 17:22:56

标签: python parameters decorator optional-parameters python-decorators

我想为bar函数定义参数param2的默认值:

def foo(default_value):
    def bar(param1, param2=default_value)
        """bar does some stuff"""
    return bar
bar = foo(default_value)

装饰师怎么可能这样呢?

更全局,如果某人有一些关于如何在使用参数装饰函数时操纵args和kwargs的文档,我很乐意看看。

2 个答案:

答案 0 :(得分:2)

带参数的装饰器的语法不同。它应该返回一个函数,它将接受一个函数并返回另一个函数。例如:

def foo(default_value):
    def wrapped(f):
        def bar(param1, param2=default_value):
            print "inside bar"
            print "param1=%s param%s"%(param1, param2)
        return bar
    return wrapped

现在当你装饰一个函数时:

@foo(default_value="test param2")
def test_decorator():
    pass

调用它将导致:

test_decorator("test param1")

inside bar
param1=test param1 param2=test param2

或者不接受默认值:

test_decorator("test param1", "other test param2")

inside bar
param1=test param1 param2=other test param2

您可以阅读有关主题here

的更多信息

答案 1 :(得分:0)

我不确定我理解您的示例代码的含义,但以下内容似乎与您的问题标题有关:

PLACE_HOLDER = object()

def default_arg(default_value):
    def decorator(f):
        if f.func_defaults and f.func_defaults[0] is PLACE_HOLDER:
            f.func_defaults = (default_value,)
        return f
    return decorator

@default_arg(42)
def bar(param1, param2=PLACE_HOLDER):
    """bar does some stuff"""
    print('bar({}, {}) called'.format(param1, param2))


bar(1)

输出:

bar(1, 42) called