Add exception handling

DSM-DMS · Mar 3, 2022 · 07d1a2e · 07d1a2e
1 parent 34eafa6
commit 07d1a2e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Server/app/__init__.py b/Server/app/__init__.py
@@ -22,6 +22,7 @@ def register_hook(flask_app: Flask):
     from app.hook.exception_handler import http_exception_handler
     flask_app.register_error_handler(JWTExtendedException, jwt_handle)
     flask_app.register_error_handler(PyJWTError, jwt_handle)
+    flask_app.register_error_handler(exception.DMSException, http_exception_handler)
 
 
 def register_blueprint(flask_app: Flask):