实际函数参数的数量

Question

有没有办法获得传递给函数的实际参数的数量

def foo(a, optional=42):
    if ????
        print "1 arg"
    else:
        print "2 args"

foo(11)      # should print 1 arg
foo(22, 42)  # should print 2 args

没有改变其签名接受*args？

Answer 1

您可以将默认值更改为sentinel:

_sentinel = object()

def foo(a, optional=_sentinel):
    if optional is _sentinel:
        optional = 42
        print "1 arg"
    else:
        print "2 args"

或者直接在func_defaults元组中访问它:

def foo(a, optional=object()):
    if optional is foo.func_defaults[0]:
        optional = 42
        print "1 arg"
    else:
        print "2 args"

但实际上并没有使用它; 这只会让那些不熟悉标准函数对象属性的人感到困惑.

是的,该_sentinel对象是内省的,并且仍然可以由一个坚定的开发人员获得,但是再一次,同一个开发人员也可以只是对你的函数进行monkeypatch.:-)

Answer 2

你可以使用装饰器来做到这一点.从保存装饰函数的签名问题我们知道如何正确地执行此操作.

import decorator


@decorator.decorator
def count_args(f):
    def new(*args, **kwargs):
        kwargs['nargs'] = len(args)+len(kwargs)
        return f(*args, **kwargs)
    return new

@count_args
def foo(a, optional=42, nargs=1):
    print nargs

foo(1) # 1
foo(1, 4) # 2
foo(1, optional=4) # 2

更新:

我刚刚为args的数量添加了关键字参数,我们将其传递给函数.如您所见,默认情况下为1,即True.这看起来像一个黑客,但它的工作原理.