Documentation logs middleware

This commit is contained in:
Yohann D'ANELLO 2020-03-11 20:36:38 +01:00
parent 1f004191a8
commit 3ad6d9e870
1 changed files with 23 additions and 1 deletions

View File

@ -14,19 +14,31 @@ _thread_locals = local()
def _set_current_user_and_ip(user=None, ip=None): def _set_current_user_and_ip(user=None, ip=None):
"""
Store current user and IP address in the local thread.
"""
setattr(_thread_locals, USER_ATTR_NAME, user) setattr(_thread_locals, USER_ATTR_NAME, user)
setattr(_thread_locals, IP_ATTR_NAME, ip) setattr(_thread_locals, IP_ATTR_NAME, ip)
def get_current_user(): def get_current_user():
"""
:return: The user that performed a request (may be anonymous)
"""
return getattr(_thread_locals, USER_ATTR_NAME, None) return getattr(_thread_locals, USER_ATTR_NAME, None)
def get_current_ip(): def get_current_ip():
"""
:return: The IP address of the user that has performed a request
"""
return getattr(_thread_locals, IP_ATTR_NAME, None) return getattr(_thread_locals, IP_ATTR_NAME, None)
def get_current_authenticated_user(): def get_current_authenticated_user():
"""
:return: The user that performed a request (must be authenticated, return None if anonymous)
"""
current_user = get_current_user() current_user = get_current_user()
if isinstance(current_user, AnonymousUser): if isinstance(current_user, AnonymousUser):
return None return None
@ -35,21 +47,31 @@ def get_current_authenticated_user():
class LogsMiddleware(object): class LogsMiddleware(object):
""" """
This middleware get the current user with his or her IP address on each request. This middleware gets the current user with his or her IP address on each request.
""" """
def __init__(self, get_response): def __init__(self, get_response):
self.get_response = get_response self.get_response = get_response
def __call__(self, request): def __call__(self, request):
"""
This function is called on each request.
:param request: The HTTP Request
:return: The HTTP Response
"""
user = request.user user = request.user
# Get request IP from the headers
# The `REMOTE_ADDR` field may not contain the true IP, if there is a proxy
if 'HTTP_X_FORWARDED_FOR' in request.META: if 'HTTP_X_FORWARDED_FOR' in request.META:
ip = request.META.get('HTTP_X_FORWARDED_FOR') ip = request.META.get('HTTP_X_FORWARDED_FOR')
else: else:
ip = request.META.get('REMOTE_ADDR') ip = request.META.get('REMOTE_ADDR')
# The user and the IP address are stored in the current thread
_set_current_user_and_ip(user, ip) _set_current_user_and_ip(user, ip)
# The request is then analysed, and the response is generated
response = self.get_response(request) response = self.get_response(request)
# We flush the connected user and the IP address for the next requests
_set_current_user_and_ip(None, None) _set_current_user_and_ip(None, None)
return response return response