python - 接受 int 或 int 元组作为 python 函数参数

Question

我有以下功能:

def port_status(self, port=(1, 2)):
    for i in port:
        print_status(i)

但我希望用户也能够像这样使用单个端口号调用函数:

port_status(port=2)

在这种特定情况下，它最终会如下所示:

TypeError: 'int' object is not iterable

我可以在这里使用一些 python 魔法吗？

Answer 1

您可以检查端口的类型。

def port_status(self, port=(1, 2)):
    if isinstance(port, int):
        port = (port,)
    for i in port:
        print_status(i)