Python如何根据类名获取类？现有实现是否有更优方案？

嘿，你的问题我太懂了——当模块里类的数量上去之后，每次调用都遍历一遍确实有点冗余，而且也不是唯一的实现方式。下面给你分享几个实用的替代方案，你可以根据自己的场景来选：

1. 提前构建类名映射字典（最适合多次查询场景） #

既然你已经用了inspect.getmembers，不如把这个遍历的工作提前做一次，而不是每次查询都重复遍历。这样后续的查询操作就是O(1)的字典查找，效率提升特别明显，尤其是需要多次调用get_class_by_name的时候：

import inspect
import my_module

# 模块加载时一次性构建映射，只执行一次
CLASS_NAME_TO_CLASS = {
    name: cls for name, cls in inspect.getmembers(my_module, inspect.isclass)
}

def get_class_by_name(input_name_string):
    # 直接通过字典获取，无需再次遍历
    return CLASS_NAME_TO_CLASS.get(input_name_string)

2. 直接利用模块的__dict__属性（更轻量的查找） #

模块里的所有成员（包括类）其实都存在它的__dict__字典里，我们可以直接通过类名字符串去查，然后再验证是不是类就行——这个方法比inspect.getmembers遍历更快，因为省去了遍历筛选的过程：

import inspect
import my_module

def get_class_by_name(input_name_string):
    # 直接从模块的命名空间取对象
    candidate = my_module.__dict__.get(input_name_string)
    # 验证是不是类，避免把同名的函数/变量当成类返回
    if inspect.isclass(candidate):
        return candidate
    return None

3. 让类自动注册到注册表（最适合动态新增类的场景） #

如果你的模块会持续新增类，或者你不想每次加新类都手动更新映射，那可以给类加一个自动注册的逻辑。比如定义一个基类，让所有需要被查找的类继承它，利用Python的__init_subclass__魔法方法自动完成注册：

# 定义全局注册表
CLASS_REGISTRY = {}

class RegistrableBase:
    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        # 把子类的名字和类本身存入注册表
        CLASS_REGISTRY[cls.__name__] = cls

# 你的业务类只需要继承这个基类就行
class User(RegistrableBase):
    pass

class Order(RegistrableBase):
    pass

def get_class_by_name(input_name_string):
    return CLASS_REGISTRY.get(input_name_string)

这种方式完全不需要依赖inspect，而且新增类的时候不用额外操作，注册表会自动更新，特别省心。

总结一下 #

• 如果只是偶尔查询类，你原来的inspect遍历方法完全够用；
• 如果需要频繁查询，优先选提前构建映射字典的方案；
• 想要更轻量的实现，就直接用模块的__dict__查找；
• 如果模块会持续新增类，自动注册的方式是最优雅的选择。

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