mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-10 04:50:16 +02:00
dc4b4bbdf5
Prevents syncing of old groups when celery workers execute faster than db transactions can be completed.
105 lines
4.1 KiB
Python
105 lines
4.1 KiB
Python
from __future__ import unicode_literals
|
|
from django.db import transaction
|
|
from django.db.models.signals import m2m_changed
|
|
from django.db.models.signals import post_save
|
|
from django.db.models.signals import pre_save
|
|
from django.db.models.signals import post_delete
|
|
from django.db.models.signals import pre_delete
|
|
from django.dispatch import receiver
|
|
from django.contrib.auth.models import User
|
|
import logging
|
|
from services.tasks import update_jabber_groups
|
|
from services.tasks import update_mumble_groups
|
|
from services.tasks import update_forum_groups
|
|
from services.tasks import update_ipboard_groups
|
|
from services.tasks import update_discord_groups
|
|
from services.tasks import update_teamspeak3_groups
|
|
from services.tasks import update_discourse_groups
|
|
from services.tasks import update_smf_groups
|
|
from authentication.tasks import set_state
|
|
from authentication.tasks import disable_member
|
|
from authentication.models import AuthServicesInfo
|
|
from services.models import AuthTS
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@receiver(m2m_changed, sender=User.groups.through)
|
|
def m2m_changed_user_groups(sender, instance, action, *args, **kwargs):
|
|
logger.debug("Received m2m_changed from %s groups with action %s" % (instance, action))
|
|
|
|
def trigger_service_group_update():
|
|
logger.debug("Triggering service group update for %s" % instance)
|
|
auth, c = AuthServicesInfo.objects.get_or_create(user=instance)
|
|
if auth.jabber_username:
|
|
update_jabber_groups.delay(instance.pk)
|
|
if auth.teamspeak3_uid:
|
|
update_teamspeak3_groups.delay(instance.pk)
|
|
if auth.forum_username:
|
|
update_forum_groups.delay(instance.pk)
|
|
if auth.smf_username:
|
|
update_smf_groups.delay(instance.pk)
|
|
if auth.ipboard_username:
|
|
update_ipboard_groups.delay(instance.pk)
|
|
if auth.discord_uid:
|
|
update_discord_groups.delay(instance.pk)
|
|
if auth.mumble_username:
|
|
update_mumble_groups.delay(instance.pk)
|
|
if auth.discourse_enabled:
|
|
update_discourse_groups.delay(instance.pk)
|
|
if auth.smf_username:
|
|
update_smf_groups.delay(instance.pk)
|
|
|
|
if action == "post_add" or action == "post_remove" or action == "post_clear":
|
|
logger.debug("Waiting for commit to trigger service group update for %s" % instance)
|
|
transaction.on_commit(trigger_service_group_update)
|
|
|
|
|
|
def trigger_all_ts_update():
|
|
for auth in AuthServicesInfo.objects.filter(teamspeak3_uid__isnull=False):
|
|
update_teamspeak3_groups.delay(auth.user.pk)
|
|
|
|
|
|
@receiver(m2m_changed, sender=AuthTS.ts_group.through)
|
|
def m2m_changed_authts_group(sender, instance, action, *args, **kwargs):
|
|
logger.debug("Received m2m_changed from %s ts_group with action %s" % (instance, action))
|
|
if action == "post_add" or action == "post_remove":
|
|
trigger_all_ts_update()
|
|
|
|
|
|
@receiver(post_save, sender=AuthTS)
|
|
def post_save_authts(sender, instance, *args, **kwargs):
|
|
logger.debug("Received post_save from %s" % instance)
|
|
trigger_all_ts_update()
|
|
|
|
|
|
@receiver(post_delete, sender=AuthTS)
|
|
def post_delete_authts(sender, instance, *args, **kwargs):
|
|
logger.debug("Received post_delete signal from %s" % instance)
|
|
trigger_all_ts_update()
|
|
|
|
|
|
@receiver(pre_delete, sender=User)
|
|
def pre_delete_user(sender, instance, *args, **kwargs):
|
|
logger.debug("Received pre_delete from %s" % instance)
|
|
disable_member(instance)
|
|
|
|
|
|
@receiver(pre_save, sender=User)
|
|
def pre_save_user(sender, instance, *args, **kwargs):
|
|
logger.debug("Received pre_save from %s" % instance)
|
|
# check if user is being marked active/inactive
|
|
if not instance.pk:
|
|
# new model being created
|
|
return
|
|
try:
|
|
old_instance = User.objects.get(pk=instance.pk)
|
|
if old_instance.is_active and not instance.is_active:
|
|
logger.info("Disabling services for inactivation of user %s" % instance)
|
|
disable_member(instance)
|
|
elif instance.is_active and not old_instance.is_active:
|
|
logger.info("Assessing state of reactivated user %s" % instance)
|
|
set_state(instance)
|
|
except User.DoesNotExist:
|
|
pass
|