Skip to content

Commit a0a17df

Browse files
committed
Remove checks for who wrote response (not necessary)
1 parent 32cdce1 commit a0a17df

1 file changed

Lines changed: 1 addition & 5 deletions

File tree

aikido_zen/sources/flask.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,15 +26,11 @@ def _full_dispatch_request_before(func, instance, args, kwargs):
2626
if not pre_response:
2727
return None
2828
# This happens when a route is rate limited, a user blocked, etc...
29-
res = Response(pre_response[0], status=pre_response[1], mimetype="text/plain")
30-
res.is_aikido_response = True # Use this in the @after
31-
return res
29+
return Response(pre_response[0], status=pre_response[1], mimetype="text/plain")
3230

3331

3432
@after
3533
def _full_dispatch_request_after(func, instance, args, kwargs, return_value):
36-
if hasattr(return_value, "is_aikido_response") and return_value.is_aikido_response:
37-
return # We wrote this response, so the status code is not reflective of the route
3834
if not hasattr(return_value, "status_code"):
3935
return
4036
funcs.request_handler(stage="post_response", status_code=return_value.status_code)

0 commit comments

Comments
 (0)