custom errors instead of default errors

Author: ksh7

api/__init__.py

@@ -3,6 +3,8 @@
 Copyright (c) 2019 - present AppSeed.us
 """
 
+import json
+
 from flask import Flask
 
 from .routes import rest_api, jwt
@@ -15,3 +17,24 @@
 db.init_app(app)
 rest_api.init_app(app)
 jwt.init_app(app)
+
+
+"""
+   Custom responses
+"""
+
+@app.after_request
+def after_request(response):
+    """
+       Sends back a custom error with {"success", "msg"} format
+    """
+
+    if int(response.status_code) >= 400:
+        response_data = json.loads(response.get_data())
+        if "errors" in response_data:
+            _err_keys = response_data.keys()
+            response_data = {"success": False,
+                             "msg": list(response_data["errors"].items())[0][1]}
+            response.set_data(json.dumps(response_data))
+        response.headers.add('Content-Type', 'application/json')
+    return response