如何使用带有multipledispatch的默认参数?

时间:2019-01-10 16:07:37

标签: python python-3.x multiple-dispatch

我正在使用multipledispatch(基于this answer)重载默认参数的函数

from multipledispatch import dispatch

class TestExampleTest(AbstractTest):
    @dispatch(ClassOne, bool, bool)
    def function(self, my_class, a=True, b=True):
        do_something()

    @dispatch(ClassTwo, bool)
    def function(self, my_class, a=True):
        do_something_else()

当我调用function()而没有将值传递给bool项的时候

self.function(ClassOne())

我明白了

  

NotImplementedError:找不到函数的签名

完整的堆栈跟踪:

ExampleTest.py:27 (TestExampleTest.test_example_test)
self = <ExampleTest.TestExampleTest object at 0x04326BB0>

    def test_example_test(self):
>       self.function(ClassOne())

ExampleTest.py:29: 
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _

self = <dispatched function>
args = (<ExampleTest.ClassOne object at 0x043262B0>,), kwargs = {}
types = (<class 'ExampleTest.ClassOne'>,), func = None

    def __call__(self, *args, **kwargs):
        types = tuple([type(arg) for arg in args])
        func = self.dispatch(*types)
        if not func:
            raise NotImplementedError('Could not find signature for %s: <%s>' %
>                                     (self.name, str_signature(types)))
E           NotImplementedError: Could not find signature for function: <ClassOne>

..\..\..\..\Automation\lib\site-packages\multipledispatch\dispatcher.py:434: NotImplementedError

注意:我知道我可以一起放@dispatch并做类似的事情

def function(self, my_class_one=None, my_class_two=None, a=True, b=True):
    if my_class_one:
        do_something()
    elif my_class_two:
        do_something_else()

但是我想知道是否可以保留当前结构。

我该如何解决?

1 个答案:

答案 0 :(得分:0)

默认参数应在@dispatch中声明为关键字参数,然后记住,如果要在调用这些函数时修改具有默认值的参数值,请传递关键字而不是依赖位置。

from multipledispatch import dispatch

class TestExampleTest(AbstractTest):
    @dispatch(ClassOne, a=bool, b=bool)
    def function(self, my_class, a=True, b=True):  # 1st function
        do_something()

    @dispatch(ClassTwo, a=bool)
    def function(self, my_class, a=True):          # 2nd function
        do_something_else()


test_ins = TestExampleTest()

# call 1st function
test_ins.function(class_one_ins) 
test_ins.function(class_one_ins, a=False)
test_ins.function(class_one_ins, a=False, b=False)

# call 2nd function
test_ins.function(class_two_ins)
test_ins.function(class_two_ins, a=False)