nk20/apps/logs/signals.py

100 lines
2.9 KiB
Python
Raw Normal View History

2020-02-24 17:18:44 +00:00
# Copyright (C) 2018-2020 by BDE ENS Paris-Saclay
# SPDX-License-Identifier: GPL-3.0-or-later
import inspect
from django.contrib.contenttypes.models import ContentType
2020-02-24 17:18:44 +00:00
from django.core import serializers
from django.db.models.signals import pre_save, pre_delete
from django.dispatch import receiver
from .models import Changelog
2020-02-27 13:47:34 +00:00
def get_request_in_signal(sender):
2020-02-27 12:34:38 +00:00
req = None
2020-02-24 17:18:44 +00:00
for entry in reversed(inspect.stack()):
try:
2020-02-27 12:34:38 +00:00
req = entry[0].f_locals['request']
# Check if there is a user
# noinspection PyStatementEffect
req.user
2020-02-24 17:18:44 +00:00
break
except:
pass
2020-02-27 12:34:38 +00:00
if not req:
2020-02-24 17:18:44 +00:00
print("WARNING: Attempt to save " + str(sender) + " with no user")
2020-02-27 12:34:38 +00:00
return req
2020-02-24 17:18:44 +00:00
2020-02-27 13:47:34 +00:00
def get_user_and_ip(sender):
req = get_request_in_signal(sender)
try:
user = req.user
if 'HTTP_X_FORWARDED_FOR' in req.META:
ip = req.META.get('HTTP_X_FORWARDED_FOR')
else:
ip = req.META.get('REMOTE_ADDR')
except:
user = None
ip = None
return user, ip
2020-02-24 17:18:44 +00:00
EXCLUDED = [
2020-02-27 14:30:16 +00:00
'admin.logentry',
'authtoken.token',
'cas_server.user',
'cas_server.userattributes',
'contenttypes.contenttype',
'logs.changelog',
'migrations.migration',
'note.noteuser',
'note.noteclub',
'note.notespecial',
'sessions.session',
'reversion.revision',
'reversion.version',
2020-02-24 17:18:44 +00:00
]
@receiver(pre_save)
def save_object(sender, instance, **kwargs):
2020-02-27 14:30:16 +00:00
# noinspection PyProtectedMember
if instance._meta.label_lower in EXCLUDED:
2020-02-24 17:18:44 +00:00
return
previous = sender.objects.filter(pk=instance.pk).all()
2020-02-27 13:47:34 +00:00
user, ip = get_user_and_ip(sender)
2020-02-27 12:34:38 +00:00
2020-02-27 13:59:01 +00:00
previous_json = serializers.serialize('json', previous)[1:-1] if previous.exists else None
instance_json = serializers.serialize('json', [instance, ])[1:-1]
2020-02-24 17:18:44 +00:00
Changelog.objects.create(user=user,
2020-02-27 13:47:34 +00:00
ip=ip,
model=ContentType.objects.get_for_model(instance),
instance_pk=instance.pk,
previous=previous_json,
data=instance_json,
action=("edit" if previous.exists() else "create")
).save()
2020-02-24 17:18:44 +00:00
2020-02-27 12:34:38 +00:00
2020-02-24 17:18:44 +00:00
@receiver(pre_delete)
def delete_object(sender, instance, **kwargs):
2020-02-27 14:30:16 +00:00
# noinspection PyProtectedMember
if instance._meta.label_lower in EXCLUDED:
2020-02-24 17:18:44 +00:00
return
2020-02-27 13:47:34 +00:00
user, ip = get_user_and_ip(sender)
2020-02-27 12:34:38 +00:00
2020-02-24 17:18:44 +00:00
instance_json = serializers.serialize('json', [instance, ])[1:-1]
Changelog.objects.create(user=user,
2020-02-27 13:47:34 +00:00
ip=ip,
model=ContentType.objects.get_for_model(instance),
instance_pk=instance.pk,
previous=instance_json,
data=None,
action="delete"
).save()