添加无参函数droplets到命令字典调用时触发TypeError问题咨询

你遇到的问题很典型——当前代码不管目标函数是否需要参数，都会把拆分后的args列表传递过去，而droplets()定义为不接受任何参数，所以调用时自然会触发TypeError。

这里有几种实用的解决方案，你可以根据自己的需求选择：

方案1：让无参函数兼容任意参数（最简单） #

直接修改droplets函数，让它接受可变参数（即使你用不到这些参数），这样不管代码传什么过来都不会报错：

from functools import partial
def create_droplet(args):
    print(args)
def droplets(*args, **kwargs):  # 添加可变参数，兼容传入的参数
    print("test")
commands = {
    'create_droplet': partial(create_droplet),
    'droplets': droplets,  # 直接添加函数引用
}
command_in = input()
command = command_in.split(" ")[:1][0]
if command in commands:
    args = command_in.split(" ")[1:]
    commands[command](args)

方案2：用Lambda包装无参函数（无需修改原函数） #

如果你不想改动droplets的原始定义，可以在commands字典里用lambda来包装它，主动忽略传入的参数：

from functools import partial
def create_droplet(args):
    print(args)
def droplets():
    print("test")
commands = {
    'create_droplet': partial(create_droplet),
    'droplets': lambda _: droplets(),  # Lambda接收参数但不传递给原函数
}
command_in = input()
command = command_in.split(" ")[:1][0]
if command in commands:
    args = command_in.split(" ")[1:]
    commands[command](args)

方案3：动态判断函数参数（最通用） #

如果你的命令字典里有很多不同参数需求的函数，可以用inspect模块动态判断函数的参数数量，然后决定是否传递args：

from functools import partial
import inspect

def create_droplet(args):
    print(args)
def droplets():
    print("test")
commands = {
    'create_droplet': partial(create_droplet),
    'droplets': droplets,
}
command_in = input()
command = command_in.split(" ")[:1][0]
if command in commands:
    func = commands[command]
    # 获取函数的签名信息，判断参数数量
    sig = inspect.signature(func)
    if len(sig.parameters) > 0:
        args = command_in.split(" ")[1:]
        func(args)
    else:
        func()  # 无参函数直接调用

各方案适用场景 #

• 方案1适合快速修改，函数本身不需要严格的参数校验；
• 方案2适合不想改动原函数代码的场景；
• 方案3适合命令数量多、参数需求复杂的情况，扩展性更强。

内容的提问来源于stack exchange，提问作者Jordan Baron