假设我有一个类Suit和四个子类Suit:红心、黑桃、方块、梅花。
class Suit:
...
class Heart(Suit):
...
class Spade(Suit):
...
class Diamond(Suit):
...
class Club(Suit):
...
我有一个方法,它接收suit作为参数,这是一个类对象,而不是一个实例。更准确地说,它可能只接受四个值中的一个:红心,黑桃,方块,梅花。我怎么能做出保证这样的断言呢?喜欢的东西:
def my_method(suit):
assert(suit subclass of Suit)
...
我用的是Python 3。
使用is子类似乎是一种编写日志级别的干净方法。用它感觉有点奇怪……但它似乎比其他选择更干净。
class Error(object): pass
class Warn(Error): pass
class Info(Warn): pass
class Debug(Info): pass
class Logger():
LEVEL = Info
@staticmethod
def log(text,level):
if issubclass(Logger.LEVEL,level):
print(text)
@staticmethod
def debug(text):
Logger.log(text,Debug)
@staticmethod
def info(text):
Logger.log(text,Info)
@staticmethod
def warn(text):
Logger.log(text,Warn)
@staticmethod
def error(text):
Logger.log(text,Error)
根据Python文档,我们也可以使用类。__mro__属性或class.mro()方法:
class Suit:
pass
class Heart(Suit):
pass
class Spade(Suit):
pass
class Diamond(Suit):
pass
class Club(Suit):
pass
>>> Heart.mro()
[<class '__main__.Heart'>, <class '__main__.Suit'>, <class 'object'>]
>>> Heart.__mro__
(<class '__main__.Heart'>, <class '__main__.Suit'>, <class 'object'>)
Suit in Heart.mro() # True
object in Heart.__mro__ # True
Spade in Heart.mro() # False