Check if a exception has __cause__ before trying to access it

darrenburns · Oct 21, 2019 · 0fd1eb5 · 0fd1eb5
1 parent 781537d
commit 0fd1eb5
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/ward/terminal.py b/ward/terminal.py
@@ -153,7 +153,10 @@ def print_traceback(self, err):
         trace = getattr(err, "__traceback__", "")
         if trace:
             trc = traceback.format_exception(None, err, trace)
-
+            if err.__cause__:
+                cause = err.__cause__.__class__.__name__
+            else:
+                cause = None
             for line in trc:
                 sublines = line.split("\n")
                 for subline in sublines:
@@ -163,7 +166,7 @@ def print_traceback(self, err):
                     elif subline.lstrip().startswith("Traceback"):
                         cprint(content, color="blue")
                     elif (subline.lstrip().startswith(err.__class__.__name__) or
-                          subline.lstrip().startswith(err.__cause__.__class__.__name__)):
+                          (cause and subline.lstrip().startswith(cause))):
                         cprint(content, color="blue")
                     else:
                         print(content)