Misago/misago/users/signals.py

from datetime import timedelta

from django.contrib.auth import get_user_model
from django.db.models import Q
from django.dispatch import Signal, receiver
from django.utils import timezone

from misago.conf import settings
from misago.core.utils import ANONYMOUS_IP

from .models import AuditTrail


UserModel = get_user_model()

anonymize_user_content = Signal()
delete_user_content = Signal()
remove_old_ips = Signal()
username_changed = Signal()


@receiver(username_changed)
def handle_name_change(sender, **kwargs):
    sender.user_renames.update(changed_by_username=sender.username)


@receiver(remove_old_ips)
def anonymize_old_registrations_ips(sender, **kwargs):
    anonymization_cutoff = timezone.now() - timedelta(days=settings.MISAGO_IP_STORE_TIME)
    ip_is_too_new = Q(joined_on__gt=anonymization_cutoff)
    ip_is_already_anonymized = Q(joined_from_ip=ANONYMOUS_IP)
    
    queryset = UserModel.objects.exclude(ip_is_too_new | ip_is_already_anonymized)
    queryset.update(joined_from_ip=ANONYMOUS_IP)


@receiver(remove_old_ips)
def remove_old_audit_trails(sender, **kwargs):
    removal_cutoff = timezone.now() - timedelta(days=settings.MISAGO_IP_STORE_TIME)
    AuditTrail.objects.filter(created_at__lte=removal_cutoff).delete()