import getpass from django.contrib.auth.models import User from django.contrib.contenttypes.models import ContentType from rest_framework.renderers import JSONRenderer from rest_framework.serializers import ModelSerializer from tfjm.middlewares import get_current_authenticated_user, get_current_ip from .models import Changelog # Ces modèles ne nécessitent pas de logs EXCLUDED = [ 'admin.logentry', 'authtoken.token', 'contenttypes.contenttype', 'logs.changelog', # Never remove this line 'mailer.dontsendentry', 'mailer.message', 'mailer.messagelog', 'migrations.migration', 'sessions.session', ] def pre_save_object(sender, instance, **kwargs): """ Before a model get saved, we get the previous instance that is currently in the database """ qs = sender.objects.filter(pk=instance.pk).all() if qs.exists(): instance._previous = qs.get() else: instance._previous = None def save_object(sender, instance, raw, **kwargs): """ Each time a model is saved, an entry in the table `Changelog` is added in the database in order to store each modification made """ # noinspection PyProtectedMember if instance._meta.label_lower in EXCLUDED or hasattr(instance, "_no_signal") or raw: return # noinspection PyProtectedMember previous = instance._previous # 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() 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 ip = "127.0.0.1" username = getpass.getuser() user = User.objects.get(username=username) if User.objects.filter(username=username).exists() else None # On n'enregistre pas les connexions # noinspection PyProtectedMember if user is not None and instance._meta.label_lower == "auth.user" and previous \ and instance.last_login != previous.last_login: return changed_fields = '__all__' if previous: # On ne garde que les champs modifiés changed_fields = [] for field in instance._meta.fields: if field.name.endswith("_ptr"): # A field ending with _ptr is a OneToOneRel with a subclass, e.g. NoteClub.note_ptr -> Note continue if getattr(instance, field.name) != getattr(previous, field.name): changed_fields.append(field.name) if len(changed_fields) == 0: # Pas de log s'il n'y a pas de modification return # On crée notre propre sérialiseur JSON pour pouvoir sauvegarder les modèles avec uniquement les champs modifiés class CustomSerializer(ModelSerializer): class Meta: model = instance.__class__ fields = changed_fields previous_json = JSONRenderer().render(CustomSerializer(previous).data).decode("UTF-8") if previous else "" instance_json = JSONRenderer().render(CustomSerializer(instance).data).decode("UTF-8") Changelog.objects.create(user=user, 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") ).save() def delete_object(sender, instance, **kwargs): """ Each time a model is deleted, an entry in the table `Changelog` is added in the database """ # noinspection PyProtectedMember if instance._meta.label_lower in EXCLUDED or hasattr(instance, "_no_signal"): 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() 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 ip = "127.0.0.1" username = getpass.getuser() user = User.objects.get(username=username) if User.objects.filter(username=username).exists() else None # 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") Changelog.objects.create(user=user, ip=ip, model=ContentType.objects.get_for_model(instance), instance_pk=instance.pk, previous=instance_json, data="", action="delete" )