django-axes/axes/decorators.py
Aleksi Häkli 10f6e621ce Add missing @wraps decorator
This keeps the original function name and docstring for the decorated function
2021-01-05 11:57:53 +02:00

26 lines
628 B
Python

from functools import wraps
from axes.handlers.proxy import AxesProxyHandler
from axes.helpers import get_lockout_response
def axes_dispatch(func):
@wraps(func)
def inner(request, *args, **kwargs):
if AxesProxyHandler.is_allowed(request):
return func(request, *args, **kwargs)
return get_lockout_response(request)
return inner
def axes_form_invalid(func):
@wraps(func)
def inner(self, *args, **kwargs):
if AxesProxyHandler.is_allowed(self.request):
return func(self, *args, **kwargs)
return get_lockout_response(self.request)
return inner