我有以下功能:
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 魔法吗?
请您参考如下方法:
您可以检查端口的类型。
def port_status(self, port=(1, 2)):
if isinstance(port, int):
port = (port,)
for i in port:
print_status(i)
