mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2025-07-10 13:00:16 +02:00
8987cf2199
Use django-redis-cache backend for locking get_or_set No longer require group-related tasks to be locked to one simultaneous execution. Remove legacy service group cache models. Truncate Discord nicknames to 32 characters Correct Discourse group name extension using only valid leading characters. Prevent name slicing from ending with illegal character Closes #801 Closes #847 Closes #835 Closes #852
129 lines
4.8 KiB
Python
129 lines
4.8 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import logging
|
|
|
|
from alliance_auth.celeryapp import app
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import User
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
|
|
from eveonline.managers import EveManager
|
|
from notifications import notify
|
|
from services.modules.discord.manager import DiscordOAuthManager, DiscordApiBackoff
|
|
from .models import DiscordUser
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class DiscordTasks:
|
|
def __init__(self):
|
|
pass
|
|
|
|
@classmethod
|
|
def add_user(cls, user, code):
|
|
user_id = DiscordOAuthManager.add_user(code)
|
|
if user_id:
|
|
discord_user = DiscordUser()
|
|
discord_user.user = user
|
|
discord_user.uid = user_id
|
|
discord_user.save()
|
|
if settings.DISCORD_SYNC_NAMES:
|
|
cls.update_nickname.delay(user.pk)
|
|
cls.update_groups.delay(user.pk)
|
|
return True
|
|
return False
|
|
|
|
@classmethod
|
|
def delete_user(cls, user, notify_user=False):
|
|
if cls.has_account(user):
|
|
logger.debug("User %s has discord account %s. Deleting." % (user, user.discord.uid))
|
|
if DiscordOAuthManager.delete_user(user.discord.uid):
|
|
user.discord.delete()
|
|
if notify_user:
|
|
notify(user, 'Discord Account Disabled', level='danger')
|
|
return True
|
|
return False
|
|
|
|
@classmethod
|
|
def has_account(cls, user):
|
|
"""
|
|
Check if the user has an account (has a DiscordUser record)
|
|
:param user: django.contrib.auth.models.User
|
|
:return: bool
|
|
"""
|
|
try:
|
|
user.discord
|
|
except ObjectDoesNotExist:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
@staticmethod
|
|
@app.task(bind=True, name='discord.update_groups')
|
|
def update_groups(task_self, pk):
|
|
user = User.objects.get(pk=pk)
|
|
logger.debug("Updating discord groups for user %s" % user)
|
|
if DiscordTasks.has_account(user):
|
|
groups = []
|
|
for group in user.groups.all():
|
|
groups.append(str(group.name))
|
|
if len(groups) == 0:
|
|
logger.debug("No syncgroups found for user. Adding empty group.")
|
|
groups.append('empty')
|
|
logger.debug("Updating user %s discord groups to %s" % (user, groups))
|
|
try:
|
|
DiscordOAuthManager.update_groups(user.discord.uid, groups)
|
|
except DiscordApiBackoff as bo:
|
|
logger.info("Discord group sync API back off for %s, "
|
|
"retrying in %s seconds" % (user, bo.retry_after))
|
|
raise task_self.retry(countdown=bo.retry_after)
|
|
except Exception as e:
|
|
if task_self:
|
|
logger.exception("Discord group sync failed for %s, retrying in 10 mins" % user)
|
|
raise task_self.retry(countdown=60 * 10)
|
|
else:
|
|
# Rethrow
|
|
raise e
|
|
logger.debug("Updated user %s discord groups." % user)
|
|
else:
|
|
logger.debug("User does not have a discord account, skipping")
|
|
|
|
@staticmethod
|
|
@app.task(name='discord.update_all_groups')
|
|
def update_all_groups():
|
|
logger.debug("Updating ALL discord groups")
|
|
for discord_user in DiscordUser.objects.exclude(uid__exact=''):
|
|
DiscordTasks.update_groups.delay(discord_user.user.pk)
|
|
|
|
@staticmethod
|
|
@app.task(bind=True, name='discord.update_nickname')
|
|
def update_nickname(self, pk):
|
|
user = User.objects.get(pk=pk)
|
|
logger.debug("Updating discord nickname for user %s" % user)
|
|
if DiscordTasks.has_account(user):
|
|
character = EveManager.get_main_character(user)
|
|
logger.debug("Updating user %s discord nickname to %s" % (user, character.character_name))
|
|
try:
|
|
DiscordOAuthManager.update_nickname(user.discord.uid, character.character_name)
|
|
except Exception as e:
|
|
if self:
|
|
logger.exception("Discord nickname sync failed for %s, retrying in 10 mins" % user)
|
|
raise self.retry(countdown=60 * 10)
|
|
else:
|
|
# Rethrow
|
|
raise e
|
|
logger.debug("Updated user %s discord nickname." % user)
|
|
else:
|
|
logger.debug("User %s does not have a discord account" % user)
|
|
|
|
@staticmethod
|
|
@app.task(name='discord.update_all_nicknames')
|
|
def update_all_nicknames():
|
|
logger.debug("Updating ALL discord nicknames")
|
|
for discord_user in DiscordUser.objects.exclude(uid__exact=''):
|
|
DiscordTasks.update_nickname.delay(discord_user.user.pk)
|
|
|
|
@classmethod
|
|
def disable(cls):
|
|
DiscordUser.objects.all().delete()
|