fix: Added error handler for status 429

app/__init__.py

@@ -137,6 +137,7 @@ def create_app():
     # development api
     with app.app_context():
         from app.api.admin_statistics_api.events import event_statistics
+        from app.api.helpers.errors import error_blueprint
         from app.api.auth import auth_routes
         from app.api.attendees import attendee_misc_routes
         from app.api.bootstrap import api_v1
@@ -177,6 +178,7 @@ def create_app():
         app.register_blueprint(ticket_blueprint)
         app.register_blueprint(order_blueprint)
         app.register_blueprint(event_blueprint)
+        app.register_blueprint(error_blueprint)
 
         add_engine_pidguard(db.engine)
 

app/api/helpers/errors.py

@@ -1,8 +1,9 @@
 import json
 
-from flask import make_response
+from flask import make_response, Blueprint
 from flask_rest_jsonapi.errors import jsonapi_errors
 
+error_blueprint = Blueprint('errors', __name__)
 
 class ErrorResponse:
     """
@@ -78,3 +79,16 @@ class BadRequestError(ErrorResponse):
     """
     status = 400
     title = 'Bad Request'
+
+
+class RequestLimitError(ErrorResponse):
+    """
+    Default class for 429 Error
+    """
+    status = 429
+    title = 'Request Limit Exceeded'
+
+
+@error_blueprint.app_errorhandler(429)
+def handle_429(error):
+    raise RequestLimitError({'source': ''}, error.description).respond()

tests/all/integration/api/helpers/test_errors.py

@@ -3,7 +3,7 @@
 
 from tests.all.integration.utils import OpenEventTestCase
 from app.api.helpers.errors import ErrorResponse, ForbiddenError, NotFoundError, ServerError, \
-    UnprocessableEntityError, BadRequestError
+    UnprocessableEntityError, BadRequestError, RequestLimitError
 from tests.all.integration.setup_database import Setup
 from flask_rest_jsonapi.errors import jsonapi_errors
 from flask import make_response
@@ -47,6 +47,11 @@ def test_errors(self):
             bad_request_error = BadRequestError({'source': ''}, 'Request cannot be served')
             self.assertEqual(bad_request_error.status, 400)
 
+            # Request limit Error
+            request_limit_error = RequestLimitError({'source': ''},
+                                                    'Request limit exceeded')
+            self.assertEqual(request_limit_error.status, 429)
+
 
 if __name__ == '__main__':
     unittest.main()