如何根据对象类型获取另一个类？

Question

我有一个用例，我想根据对象类型获取另一个类。

对于一个解决方案，我创建了一个 dict，我在其中维护这种关系并使用类型在类之间进行匹配。

例如，考虑下面的代码

class classA(object):
    def __init__(self):
        print "A called"

class classB(object):
    def __init__(self):
        print "B called"

MAP = {
   A: classA,
   B: classB
}

class C(object):
    pass

class A(C):
    pass

class B(C):
    pass

obj can be object of A or B.

new_cls = MAP.get(type(obj), 'None')
new_obj = new_cls()

有没有更好的方法来实现这一点？

Answer 1

为什么不这样做：

class C(object):
    pass

class A(C):
    def call:
        print("A")

class B(C):
    def call:
        print("B")
obj.call()

或者使用继承。

但是等等，如果我不能这样做并且真的需要知道这是哪个类？？

你可以使用这个obj.__class__.__name__ 这将适用于旧风格和new style Classes

Answer 2

看起来您正在尝试重塑 type-based polymorphic dispatch...

我假设您没有使用 A 和 B 类（=> 它们属于 stdlib 或您不想分叉的第三方），否则非常明显的解决方案就是直接在这些类上实现call（或任何其他方法）。

在这种情况下，您的解决方案实际上是相当 Python 的。另一个可能的pythonic解决方案是using functools.singledispatch，最后的手段邪恶丑陋但有时是救生员的解决方案是monkeypatchA和B。