Python新手求助：如何根据用户输入调用对应类别类

嘿，作为Python新手能想到这种动态调用的需求已经超棒啦！你的问题核心是没办法直接用字符串变量去引用模块里的类——当前代码里的calculate.category会把category当成模块的属性名，而不是你存着类名的变量，所以肯定行不通。下面给你两种实用的解决思路：

方法1：用getattr()直接获取模块中的类 #

getattr()是Python内置函数，专门用来根据字符串名称获取对象的属性，这里我们可以用它从calculate模块里拿到对应名称的类：

import calculate

category = args.category
values = args.values

try:
    # 根据字符串名称获取对应的类
    TargetClass = getattr(calculate, category)
    # 创建类的实例并调用calc方法
    obj = TargetClass(values)
    obj.calc()
except AttributeError:
    # 处理类不存在的情况，避免程序崩溃
    print(f"出错啦：找不到名为 {category} 的类别类")
    # 可以根据需求添加错误处理逻辑，比如退出程序
    exit(1)

注意点： #

• 要保证用户输入的字符串（比如'travel'）和calculate.py里的类名完全一致，大小写也要匹配哦
• 每个类的__init__方法要能接收values参数，不然创建实例时会报错
• 如果values是多个参数，可能需要用*values来解包，比如obj = TargetClass(*values)

方法2：用字典做类映射（更安全可控） #

如果想限制用户只能调用你指定的类，避免输入奇怪的字符串触发意外操作，用字典映射是更稳妥的选择：

首先在主代码里创建一个“白名单”字典，把允许的输入字符串和对应的类绑定：

import calculate

# 定义允许调用的类的映射关系
category_map = {
    'travel': calculate.travel,
    'dining': calculate.dining,
    'shopping': calculate.shopping
    # 把你所有的类别类都加到这里
}

category = args.category
values = args.values

# 先检查输入是否在允许的范围内
if category not in category_map:
    print(f"抱歉，不支持 {category} 这个类别哦")
    exit(1)

# 获取对应的类并执行
TargetClass = category_map[category]
obj = TargetClass(values)
obj.calc()

这种方法的优势： #

• 完全可控，只有你加到字典里的类才能被调用，安全性更高
• 输入字符串和类名可以不一样，比如输入'trip'也可以映射到calculate.travel，灵活度更高

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