Add exception handler

DSM-DMS · Mar 2, 2022 · efdfdb1 · efdfdb1
1 parent 5e9fccb
commit efdfdb1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Server/app/__init__.py b/Server/app/__init__.py
@@ -1,7 +1,7 @@
-from werkzeug.exceptions import HTTPException
 from flask import Flask, jsonify
 from jwt.exceptions import PyJWTError
 from flask_jwt_extended.exceptions import JWTExtendedException
+from app.exception import DMSException
 
 from config import config
 
@@ -23,7 +23,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(HTTPException, http_exception_handler)
+    flask_app.register_error_handler(DMSException, http_exception_handler)
 
 
 def register_blueprint(flask_app: Flask):
@@ -58,7 +58,7 @@ def create_app(config_name: str) -> Flask:
     return flask_app
 
 
-def wrong_token_handler(wrong_token: str, _):
+def wrong_token_handler(wrong_token: str):
     return jsonify({
         'status': 403,
         'sub_status': 42,