如何为同一个函数设置不同的输入类型？

Question

我想做的基本想法是：

def aFunction(string='', dicti={}):
    if len(str) > 0:
         print('you gave string as input')
    if len(dicti) > 0:
         print('you gave a dict as input')

aFunction(string='test')
dict['test'] = test
aFunction(dicti=dict)

我知道这种想法在更多 OO 类型的语言中是可能的，但这在 Python 中也可能吗？

现在我正在做

def aFunction(input):
    if type(input) == str:
         print('you gave string as input')
    if type(input) == dict:
         print('you gave a dict as input')

aFunction('test')

但是我希望在调用函数时能够清楚地了解差异。

Answer 1

让相同的方法支持不同的参数类型的想法被称为多重调度或多方法。

要获得对其的详细介绍，您可以阅读这篇 Guido Van Rossum 文章 并查看 PyPi，因为那里有一些多方法包可用。

Answer 2

在 Python 中应避免类型检查（搜索“ducktyping” ）。当它们是必要时，通常应该使用isinstance来完成，而不是像问题所示的那样对类型进行相等检查。这样做的优点是对于继承情况更加灵活。

从 Python 3.4 开始，类似字符串的分支和类似字典的分支可以写成使用 stdlib functools.singledispatch。

因此，

def aFunction(input_):
    if isinstance(input_, str):
        print('you gave a string-like input')
        ...
    elif isinstance(input_, dict):
        print('you gave a dict-like input')
        ...

您现在可以：

from functools import singledispatch

@singledispatch
def aFunction(input_):
    pass

@aFunction.register(str)
def _(input_):
    print("you gave a string-like input")

@aFunction.register(dict)
def _(input_):
    print("you gave a dict-like input")

在 Python 3.5+ 中，您可以选择使用类型提示函数注释。阅读PEP 484 - 类型提示了解更多信息有关该功能的详细信息。这意味着单一调度 泛型函数 可以写成：

from functools import singledispatch

@singledispatch
def aFunction(input_):
    pass

@aFunction.register
def _(input_: str):
    print("you gave a string-like input")

@aFunction.register
def _(input_: dict):
    print("you gave a dict-like input")

Answer 3

你所拥有的基本上是有效的，只是变量声明存在一些问题。这是一个有效的版本：

def aFunction(str = '', dicti = {}):
    if len(str) > 0:
         print 'you gave string as input'
    if len(dicti) > 0:
         print 'you gave a dict as input'

str = 'test'
dicti = dict([('test', 'test')])
aFunction(str = str)
aFunction(dicti = dicti)
aFunction(str = str, dicti = dicti)
aFunction(dicti = dicti, str = str)

put：

you gave string as input
you gave a dict as input
you gave string as input
you gave a dict as input
you gave string as input
you gave a dict as input

您还可以将非特定参数作为列表和字典发送到函数（请参阅 什么是在 Python 中拥有多个构造函数的干净、Pythonic 方法？ 例如）。