Как использовать параметр по умолчанию с multipledispatch? - PullRequest
Новая
       61

Как использовать параметр по умолчанию с multipledispatch?

0 голосов
/ 10 января 2019

Я перегружаю функции параметром по умолчанию, используя multipledispatch (на основе этот ответ ) 

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()

Но мне интересно, смогу ли я сохранить текущую структуру.

Как я могу это исправить?

...