Hopiu/django-axes
1
0
Fork 0
mirror of https://github.com/jazzband/django-axes.git synced 2026-05-21 13:51:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Check the headers attribute on requests

Browse source 
The oauth2_provider package stores a (custom) set of headers on the
headers attribute instead of the default META attribute. Let's use it to
look up our expected keys.
This commit is contained in:
Gert Van Gool 2020-12-23 10:26:15 -08:00
parent ef5b645da2
commit 9ae2cdb168
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
axes/signals.py
View file

@ -27,7 +27,11 @@ log = logging.getLogger(settings.AXES_LOGGER)
user_locked_out = Signal(providing_args=['request', 'username', 'ip_address'])
def request_meta_get(request, key, default_value=None):
return getattr(request, 'META', {}).get(key, default_value)
meta = getattr(request, 'META', {})
if not meta:
# oauth2_provider package stores META in headers
meta = getattr(request, 'headers', {})
return meta.get(key, default_value)
@receiver(user_login_failed)

2
axes/utils.py
View file

@ -53,6 +53,8 @@ def get_ip(request):
"""Parse IP address from REMOTE_ADDR or
AXES_REVERSE_PROXY_HEADER if AXES_BEHIND_REVERSE_PROXY is set."""
request_meta = getattr(request, "META", {})
if not request_meta:
request_meta = getattr(request, "headers", {})
if settings.AXES_BEHIND_REVERSE_PROXY:
# For requests originating from behind a reverse proxy,
# resolve the IP address from the given AXES_REVERSE_PROXY_HEADER.