如果存在某个类（按字符串名称），如何检查python？

Question

是否可以检查某个类是否存在？我在 json 配置文件中有类名。 我知道我可以简单地尝试通过类名字符串创建一个对象，但这实际上是一个坏主意，因为类构造函数可以做一些意想不到的事情，而此时我只想检查我的配置是否有效并且都提到了可以上课。

有什么办法吗？

编辑：我也明白你可以从某个模块中获取所有方法，在我的情况下，我不确定并且实际上并不关心该方法来自哪个模块。它可以来自任何 import 语句，我可能不知道具体来自哪里。

Answer 1

你可以解析源码得到所有的类名：

from ast import ClassDef, parse
import importlib
import inspect

mod = "test"
mod = importlib.import_module(mod)
p = parse(inspect.getsource(mod))
names = [kls.name for kls in p.body if isinstance(kls, ClassDef)]

输入：

class Foo(object):
   pass

class Bar(object):
    pass

输出：

['Foo', 'Bar']

只需将配置中的类名与返回的名称进行比较。

{set of names in config}.difference(names)

如果您想包含导入的名称，您可以解析导入它的模块，但根据导入的方式，您仍然可以找到不起作用的情况：

from ast import ClassDef, parse, ImportFrom
import importlib
import inspect

mod = "test"
mod = importlib.import_module(mod)
p = parse(inspect.getsource(mod))
names = []
for node in p.body:
    if isinstance(node, ClassDef):
        names.append(node.name)
    elif isinstance(node, ImportFrom):
        names.extend(imp.name for imp in node.names)

print(names)

输入：

from test2 import Foobar, Barbar, foo  

class Foo(object):
   pass

class Bar(object):
    pass

测试2：

foo = 123

class Foobar(object):
    pass

class Barbar(object):
    pass

输出：

['Foobar', 'Barbar', 'Foo', 'Bar']

Answer 2

我尝试了built-in type function，它对我有用，但可能有一种更 Python 的方式来测试一个类的存在：

import types

def class_exist(className):
    result = False
    try:
        result = (eval("type("+className+")") == types.ClassType)
    except NameError:
        pass
    return result

# this is a test class, it's only purpose is pure existence:
class X: 
    pass

print class_exist('X')
print class_exist('Y')

输出是

True
False

当然，这是一个基本的解决方案，只能用于众所周知的输入：eval 函数can 是一个很好的后门开启者。有一个更可靠（但也很紧凑）solution by wenzul。

Answer 3

使用eval() 为任意代码执行敞开大门，为了安全起见，应该避免这种情况。

特别是如果您在这里要求解决此类问题。 那么我们可以假设您对这些风险没有充分了解。

import sys

def str_to_class(str):
    return reduce(getattr, str.split("."), sys.modules[__name__])

try:
    cls = str_to_class(<json-fragment-here>)
except AttributeError:
    cls = None

if cls:
    obj = cls(...)
else:
    # fight against this

这避免了使用eval，并得到了多个 SO 用户的认可。 解决方法类似于Convert string to Python class object?。