51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
# This file is part of photo21
|
|
# Copyright (C) 2022 Amicale des élèves de l'ENS Paris-Saclay
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from allauth.account.models import EmailAddress
|
|
from allauth.socialaccount.signals import pre_social_login
|
|
from django.conf import settings
|
|
from django.contrib.auth.models import Group
|
|
from django.dispatch import receiver
|
|
|
|
|
|
@receiver(pre_social_login)
|
|
def sync_user_fields(sender, request, sociallogin, **kwargs):
|
|
if not sociallogin.is_existing:
|
|
return
|
|
|
|
user = sociallogin.user
|
|
data = sociallogin.account.extra_data
|
|
changed = False
|
|
|
|
email = data.get("email")
|
|
if email and user.email != email:
|
|
user.email = email
|
|
changed = True
|
|
EmailAddress.objects.filter(user=user).update(email=email)
|
|
|
|
username = data.get("preferred_username")
|
|
if username and user.username != username:
|
|
user.username = username
|
|
changed = True
|
|
|
|
staff_groups = settings.OAUTH_STAFF_GROUPS
|
|
if staff_groups:
|
|
oauth_groups = set(data.get("groups", []))
|
|
is_staff = bool(oauth_groups & set(staff_groups))
|
|
if user.is_staff != is_staff:
|
|
user.is_staff = is_staff
|
|
changed = True
|
|
|
|
if changed:
|
|
user.save()
|
|
|
|
group_map = settings.OAUTH_GROUP_MAP
|
|
if group_map:
|
|
oauth_groups = set(data.get("groups", []))
|
|
for oauth_group, django_group_name in group_map.items():
|
|
django_group, _ = Group.objects.get_or_create(name=django_group_name)
|
|
if oauth_group in oauth_groups:
|
|
user.groups.add(django_group)
|
|
else:
|
|
user.groups.remove(django_group)
|