mirror of
https://github.com/jazzband/django-auditlog.git
synced 2026-03-16 22:20:26 +00:00
a0ae594425
* Disable logging remote IP address * Update auditlog/middleware.py * Update CHANGELOG.md * Update auditlog/middleware.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update auditlog/middleware.py and add tests in ManyRelatedModelTest * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: Hasan Ramezani <hasan.r67@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
from django.conf import settings
|
|
from django.contrib.auth import get_user_model
|
|
|
|
from auditlog.cid import set_cid
|
|
from auditlog.context import set_actor
|
|
|
|
|
|
class AuditlogMiddleware:
|
|
"""
|
|
Middleware to couple the request's user to log items. This is accomplished by currying the
|
|
signal receiver with the user from the request (or None if the user is not authenticated).
|
|
"""
|
|
|
|
def __init__(self, get_response=None):
|
|
self.get_response = get_response
|
|
if not isinstance(settings.AUDITLOG_DISABLE_REMOTE_ADDR, bool):
|
|
raise TypeError("Setting 'AUDITLOG_DISABLE_REMOTE_ADDR' must be a boolean")
|
|
|
|
@staticmethod
|
|
def _get_remote_addr(request):
|
|
if settings.AUDITLOG_DISABLE_REMOTE_ADDR:
|
|
return None
|
|
|
|
# In case there is no proxy, return the original address
|
|
if not request.headers.get("X-Forwarded-For"):
|
|
return request.META.get("REMOTE_ADDR")
|
|
|
|
# In case of proxy, set 'original' address
|
|
remote_addr: str = request.headers.get("X-Forwarded-For").split(",")[0]
|
|
|
|
# Remove port number from remote_addr
|
|
if "." in remote_addr and ":" in remote_addr: # IPv4 with port (`x.x.x.x:x`)
|
|
remote_addr = remote_addr.split(":")[0]
|
|
elif "[" in remote_addr: # IPv6 with port (`[:::]:x`)
|
|
remote_addr = remote_addr[1:].split("]")[0]
|
|
|
|
return remote_addr
|
|
|
|
@staticmethod
|
|
def _get_actor(request):
|
|
user = getattr(request, "user", None)
|
|
if isinstance(user, get_user_model()) and user.is_authenticated:
|
|
return user
|
|
return None
|
|
|
|
def __call__(self, request):
|
|
remote_addr = self._get_remote_addr(request)
|
|
user = self._get_actor(request)
|
|
|
|
set_cid(request)
|
|
|
|
with set_actor(actor=user, remote_addr=remote_addr):
|
|
return self.get_response(request)
|