fix: ApplicationException has no attribute 'body'

equinor · May 4, 2023 · 3895685 · 3895685
1 parent faad7e8
commit 3895685
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/dm_cli/dmss.py b/dm_cli/dmss.py
@@ -70,7 +70,13 @@ def dmss_exception_wrapper(
 ) -> Any:
     try:
         return function(*args, **kwargs)
-    except (ApplicationException, NotFoundException, ApiException) as e:
+    except ApplicationException as e:
+        if state.debug:
+            console.print_exception()
+        exception = e.dict()
+        console.print(exception, style="red1")
+        raise typer.Exit(code=1)
+    except (NotFoundException, ApiException) as e:
         if state.debug:
             console.print_exception()
         exception = json.loads(e.body)