From 8289df1d621bf2aab9f39c701eef74caba2cb630 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Mon, 27 Sep 2021 20:49:39 -0600
Subject: [PATCH] Don't prepend exception name for WK generic types

---
 scripts/wk/std.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/scripts/wk/std.py b/scripts/wk/std.py
index 958d275f..f6979d13 100644
--- a/scripts/wk/std.py
+++ b/scripts/wk/std.py
@@ -442,10 +442,11 @@ class TryAndPrint():
       pass
 
     # Prepend exception name
-    try:
-      message = f'{_exception.__class__.__name__}: {message}'
-    except Exception: # pylint: disable=broad-except
-      message = f'UNKNOWN ERROR: {message}'
+    if _exception.__class__.__name__ not in ('GenericError', 'GenericWarning'):
+      try:
+        message = f'{_exception.__class__.__name__}: {message}'
+      except Exception: # pylint: disable=broad-except
+        message = f'UNKNOWN ERROR: {message}'
 
     # Fix multi-line messages
     if '\n' in message:
@@ -525,8 +526,8 @@ class TryAndPrint():
       return obj
 
     # Try all modules
-    for _mod in sys.modules:
-      obj = getattr(sys.modules[_mod], name, None)
+    for _mod in sys.modules.values():
+      obj = getattr(_mod, name, None)
       if obj:
         break