mirror of https://gitlab.crans.org/bde/nk20
Documentation logs middleware
This commit is contained in:
parent
1f004191a8
commit
3ad6d9e870
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue