python - 检查类名(字符串)是否用于 Python 中的某个类

Question

如何获得(在当前范围内)在给定名称下可用的确切类变量？我想写一个这样的函数:

from my_module import ClassA # A subclass of my_other_module.BenevolentClass
from my_other_module import my_function
a = 'ClassA'
cls_var = my_function(a)
o = cls_var()

这样我就可以向 my_function 提供任何字符串，只要该字符串在调用者的命名空间中作为类名可用，它就会生成正确的类，就像我将字符串直接复制粘贴到代码中一样。原因是我需要为复杂的对象创建例程提供类名，但要避免 eval如果可能。我目前的实现是这样的:

def my_function(name):
    if name in globals():
        c = globals()[name]
        # Actually a complex class whitelist
        if issubclass(c, BenevolentClass):
            return c
        else:
            raise ValueError(f'Potentially malicious class {name}')

但这显然从 my_other_module 产生 globals()，这不是我想要的。我想要在确切的代码行中可用的所有类，其中 my_function被调用(可能在完全不同的模块中，从另一个模块中调用)。

Answer 1

您可以将全局字典传递给 my_function。

def my_function(name, g):
    if name in g:
        ...

cls_var = my_function(a, globals())