在不修改函数定义的情况下调用函数时的可选参数

2024-10-06 06:52:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我想知道如何调用一个函数,传递一个它不期望的参数。在

几天前我遇到了这个问题,并找到了一个解决办法。但今天,我决定看看我想做的事情是否可行。不幸的是,我不记得我使用它的背景。所以这里有一个愚蠢的例子,其中有很多更好的方法可以做到这一点,但请忽略它:

def test(func, arg1, arg2):
    return func(arg1, arg2, flag1=True, flag2=False) #Only pass the flags if the function accepts them.

def func1(a, b, flag1, flag2):
    ret = a
    if flag1:
        ret *= b
    if flag2:
        ret += b
    return ret

def func2(a, b):
    return a*b

print test(func1, 5, 6) #prints 30

我想到的另一种选择是这样的:

^{pr2}$

或者试一下…除了。。块

但是肯定有更好的方法在不改变func1和func2的情况下做到这一点,对吧?在

(编辑):利用Max S提供的解决方案,我认为这是最好的方法:

def callWithOptionalArgs(func, *args):
    argSpec = inspect.getargspec(func)
    lenArgSpec = len(argSpec.args or ())
    argsToPass = args[:lenArgSpec] #too many args
    defaults = argSpec.defaults or ()
    lenDefaults = len(defaults)
    argsToPass += (None, )*(lenArgSpec-len(argsToPass)-lenDefaults) #too few args
    argsToPass += defaults[len(argsToPass)+len(defaults)-lenArgSpec:] #default args

    return func(*argsToPass)
print callWithOptionalArgs(func1, 5, 6, True) #prints 30
print callWithOptionalArgs(func2, 5, 6, True) #prints 30

Tags: 方法truelenreturnifdefargsfunc
2条回答

检查函数是在不改变或修饰原始函数的情况下显式区分具有不同参数数的函数的唯一方法。我对您的包装器所做的唯一更改是将其归纳为任意数量的参数:

def padArgsWithTrue(func, *args):
    passed_args = list(args)
    num_args = len(inspect.getargspec(func).args)
    passed_args += [True] * (num_args - len(args))
    return func(*passed_args)

print padArgsWithTrue(lambda x,y,z,w: (x*y, z, w), 5, 6)

编辑:请注意,这不适用于参数或关键字参数数目可变的函数。在编写完整的解决方案之前,您必须确定一个处理这些问题的策略。在

如果我理解您的意思,您希望能够使用test调用func1和func2。在

def test(func, arg1, arg2):
    try:
        return func(arg1, arg2, flag1=True, flag2=False)
    except TypeError:
        return func(arg1, arg2)


def func1(a, b, flag1, flag2):
    ret = a
    if flag1:
        ret *= b
    if flag2:
        ret += b
    return ret

def func2(a, b):
    return a*b

print test(func1, 5, 6) #prints 30
print test(func2, 5, 6) #prints 30

相关问题 更多 >