python - 将方法作为带有可选参数的参数的方法

Question

我从库中有两种以相同方式工作的方法。不同之处在于，它需要一个附加的可选参数。例如:

def method1(a, b, c):
    ...

def method2(a, b, c, d=None):
    ...

我必须对这些方法的结果执行相同的任务，因此我有一个将它们组合起来的方法，如下所示:

def wrapper(method, a, b, c, d=None):

    ...

    if d:
        results = method(a, b, c, d=d)
    else:
        results = method(a, b, c)

    ...

这可行，但是当我添加更多具有不同可选参数的方法时，它变得很麻烦。有没有更好的方法来处理这些参数？

Answer 1

这里有一些代码可以实现您正在寻找的内容。

您可以将方法集合传递到 wrapper 中，该函数将返回任何将关键字参数映射到 kwargs 的方法的值。

def method1(a, b, c):
    return a, b, c


def method2(a, b, c, d=None):
    return a, b, c, d

methods = (
    method1,
    method2,
) # Collection of methods to run.

def wrapper(kwargs, methods=methods):
    """Loop over methods with kwargs."""
    for method in methods:
        try: # Call method with **kwargs
            return method(**kwargs) # Return value if keys in kwargs fit signature of method.
        except TypeError as err: # Handle error if keyword args don't match.
            print(f'err "{err}" for method "{method}')

kwargs_collection = (dict(zip(args, (f'value for arg: "{arg}"' for arg in args)))
                     for args in ('abcd', 'abc', ))




for test_kwargs in kwargs_collection:
    print(wrapper(test_kwargs))

输出:

err "method1() got an unexpected keyword argument 'd'" for method "function method1 at 0x7f900c2b7d90"

('value for arg: "a"', 'value for arg: "b"', 'value for arg: "c"', 'value for arg: "d"')

('value for arg: "a"', 'value for arg: "b"', 'value for arg: "c"')