Benutzerdefinierter Typ mit eingebautem Typvergleich

Question

Ich schreibe einen Kontextmanager, der es ermöglicht, einen bestimmten Typ von Ausnahme abzufangen.

class AssertRaises(object): 
    def __init__(self, exc_type): 
     self.exc_type = exc_type 

    def __enter__(self): 
     pass 

    def __exit__(self, exc_type, exc_val, exc_tb): 
     if exc_type == self.exc_type: 
      raise AssertionError 
     return True 

Dieser Manager funktioniert gut, wenn integrierte Ausnahme wirft aber nicht mit einer solchen Nutzung:

class MyTypeError(TypeError): 
    pass 


try: 
    with AssertRaises(TypeError): 
     raise MyTypeError() 
except Exception as e: 
    print(type(e).__name__) 

In diesem Beispiel wird die benutzerdefinierte excepton erhöht, aber diese Ausnahme entspricht Typeerror und ich möchte Es wird vom Kontextmanager als TypeError verarbeitet. Ich habe, dass `isinstance (MyTypeError(), Typeerror) == True‘ und

wollen in der gleichen Art und Weise

__exit__(...) 

arbeiten (Vererbung betrachten). Gibt es eine Lösung?

Answer 1

Entweder überprüfen Sie die Ausnahme selbst (exc_val), wie Sie mit isinstance() getan haben, oder verwenden Sie issubclass():

def __exit__(self, exc_type, exc_val, exc_tb): 
    if issubclass(exc_type, self.exc_type): 
     raise AssertionError 
    return True