我正在检查python 3.x中的两个变量是否属于同一类型。最理想的方法是什么?
举个例子:
class A():
def __init__(self, x):
self.x = x
class B(A):
def __init__(self, x):
x += 5
super(B, self).__init__(x)
理想情况下,如果将
True
和A
类型的两个变量相互比较,我希望返回B
。以下是一些不起作用的潜在解决方案:>>> a = A(5)
>>> b = B(5)
>>>
>>> type(a) is type(b)
False
>>> isinstance(a, type(b))
False
>>> isinstance(b, type(a))
True
最后一个并不理想,因为如中间的示例所示,如果要检查的类型是变量类型的子类,则返回
False
。我尝试过的唯一解决方案是:
>>> isinstance(a, type(b)) or isinstance(b, type(a))
True
有更好的办法吗?
最佳答案
此程序将遍历所有提供的对象,并检查它们之间的公共交集(sans__bases__
):
class A:
def __init__(self, x):
self.x = x
class B(A):
def __init__(self, x):
x += 5
super(B, self).__init__(x)
class C(B):
def __init__(self, x):
self.x = x
class D:
def __init__(self, x):
self.x = x
class E(C, B):
def __init__(self, x):
self.x = x
a = A(5)
b = B(5)
c = C(5)
d = D(5)
e = E(5)
def check(*objs):
def _all_bases(o):
for b in o.__bases__:
if b is not object:
yield b
yield from _all_bases(b)
s = [(i.__class__, *_all_bases(i.__class__)) for i in objs]
return len(set(*s[:1]).intersection(*s[1:])) > 0
print(check(a, b)) # True
print(check(a, c)) # True
print(check(a, d)) # False
print(check(a, e)) # True
print(check(b, c)) # True
print(check(b, d)) # False
print(check(b, e)) # True
print(check(e, d)) # False
print(check(a, b, c)) # True
print(check(a, b, c, e)) # True
print(check(a, b, c, d)) # False
print(check('string1', 'string2')) # True