Recursively walk subclass tree to build code->exception class mapping.

googleapis · Nov 7, 2014 · e322609 · e322609
1 parent 89788fb
commit e322609
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/gcloud/storage/exceptions.py b/gcloud/storage/exceptions.py
@@ -164,7 +164,16 @@ def make_exception(response, content):
     return error
 
 
-for name, value in globals().items():
-    code = getattr(value, 'code', None)
+def _walk_subclasses(klass):
+    """Recursively walk subclass tree."""
+    for sub in klass.__subclasses__():
+        yield sub
+        for subsub in _walk_subclasses(sub):
+            yield subsub
+
+
+# Build the code->exception class mapping.
+for eklass in _walk_subclasses(StorageError):
+    code = getattr(eklass, 'code', None)
     if code is not None:
-        _HTTP_CODE_TO_EXCEPTION[code] = value
+        _HTTP_CODE_TO_EXCEPTION[code] = eklass