nk20/apps/logs/signals.py

146 lines
5.6 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
from django.contrib.contenttypes.models import ContentType
from django.db.models.signals import pre_save, post_save, post_delete
2020-02-24 17:18:44 +00:00
from django.dispatch import receiver
2020-03-10 23:41:37 +00:00
from rest_framework.renderers import JSONRenderer
from rest_framework.serializers import ModelSerializer
import getpass
2020-03-07 21:28:59 +00:00
from note.models import NoteUser, Alias
from .middlewares import get_current_authenticated_user, get_current_ip
2020-02-24 17:18:44 +00:00
from .models import Changelog
# Ces modèles ne nécessitent pas de logs
2020-02-24 17:18:44 +00:00
EXCLUDED = [
2020-03-07 21:28:59 +00:00
'admin.logentry',
'authtoken.token',
2020-03-11 14:13:34 +00:00
'cas_server.proxygrantingticket',
'cas_server.proxyticket',
'cas_server.serviceticket',
2020-03-07 21:28:59 +00:00
'cas_server.user',
'cas_server.userattributes',
'contenttypes.contenttype',
2020-03-10 23:41:37 +00:00
'logs.changelog', # Never remove this line
2020-03-07 21:28:59 +00:00
'migrations.migration',
2020-03-10 23:41:37 +00:00
'note.note' # We only store the subclasses
'note.transaction',
2020-03-07 21:28:59 +00:00
'sessions.session',
]
2020-02-24 17:18:44 +00:00
@receiver(pre_save)
def pre_save_object(sender, instance, **kwargs):
"""
Avant la sauvegarde d'un modèle, on récupère l'ancienne instance actuellement en base de données
que l'on garde en mémoire
"""
qs = sender.objects.filter(pk=instance.pk).all()
if qs.exists():
instance._previous = qs.get()
else:
instance._previous = None
@receiver(post_save)
2020-02-24 17:18:44 +00:00
def save_object(sender, instance, **kwargs):
"""
Dès qu'un modèle est sauvegardé, une entrée dans la table `Changelog` est ajouté dans la base de données
afin de répertorier chaque modification effectuée
"""
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
# noinspection PyProtectedMember
previous = instance._previous
2020-02-27 12:34:38 +00:00
# Si un utilisateur est connecté, on récupère l'utilisateur courant ainsi que son adresse IP
user, ip = get_current_authenticated_user(), get_current_ip()
2020-03-07 15:45:45 +00:00
if user is None:
# Si la modification n'a pas été faite via le client Web, on suppose que c'est du à `manage.py`
# On récupère alors l'utilisateur·trice connecté·e à la VM, et on récupère la note associée
# IMPORTANT : l'utilisateur dans la VM doit être un des alias note du respo info
ip = "127.0.0.1"
username = Alias.normalize(getpass.getuser())
2020-03-10 23:41:37 +00:00
note = NoteUser.objects.filter(alias__normalized_name=username)
# if not note.exists():
# print("WARNING: A model attempted to be saved in the DB, but the actor is unknown: " + username)
# else:
if note.exists():
user = note.get().user
# noinspection PyProtectedMember
if user is not None and instance._meta.label_lower == "auth.user" and previous:
# On n'enregistre pas les connexions
if instance.last_login != previous.last_login:
return
2020-03-10 23:41:37 +00:00
# On crée notre propre sérialiseur JSON pour pouvoir sauvegarder les modèles
class CustomSerializer(ModelSerializer):
class Meta:
model = instance.__class__
fields = '__all__'
2020-03-11 10:15:03 +00:00
previous_json = JSONRenderer().render(CustomSerializer(previous).data).decode("UTF-8") if previous else None
instance_json = JSONRenderer().render(CustomSerializer(instance).data).decode("UTF-8")
if previous_json == instance_json:
# Pas de log s'il n'y a pas de modification
return
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 else "create")
2020-02-27 13:47:34 +00:00
).save()
2020-02-24 17:18:44 +00:00
2020-02-27 12:34:38 +00:00
@receiver(post_delete)
2020-02-24 17:18:44 +00:00
def delete_object(sender, instance, **kwargs):
"""
Dès qu'un modèle est supprimé, une entrée dans la table `Changelog` est ajouté dans la base de données
"""
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
# Si un utilisateur est connecté, on récupère l'utilisateur courant ainsi que son adresse IP
user, ip = get_current_authenticated_user(), get_current_ip()
2020-02-27 12:34:38 +00:00
if user is None:
# Si la modification n'a pas été faite via le client Web, on suppose que c'est du à `manage.py`
# On récupère alors l'utilisateur·trice connecté·e à la VM, et on récupère la note associée
# IMPORTANT : l'utilisateur dans la VM doit être un des alias note du respo info
ip = "127.0.0.1"
username = Alias.normalize(getpass.getuser())
note = NoteUser.objects.filter(alias__normalized_name=username)
# if not note.exists():
# print("WARNING: A model attempted to be saved in the DB, but the actor is unknown: " + username)
# else:
if note.exists():
user = note.get().user
2020-03-10 23:41:37 +00:00
# On crée notre propre sérialiseur JSON pour pouvoir sauvegarder les modèles
class CustomSerializer(ModelSerializer):
class Meta:
model = instance.__class__
fields = '__all__'
instance_json = JSONRenderer().render(CustomSerializer(instance).data).decode("UTF-8")
2020-03-10 23:41:37 +00:00
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=instance_json,
data=None,
action="delete"
).save()