Bugfix: Return proper error (#27)

verloop · Dec 19, 2020 · cdca925 · cdca925
1 parent 754a4f1
commit cdca925
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/twirp/exceptions.py b/twirp/exceptions.py
@@ -15,9 +15,9 @@
 
 class TwirpServerException(httplib.HTTPException):
     def __init__(self, *args, code, message, meta={}):
-        if isinstance(code, errors.Errors):
-            self._code = code
-        else:
+        try:
+            self._code = errors.Errors(code)
+        except ValueError:
             self._code = errors.Errors.Unknown
         self._message = message
         self._meta = meta