68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
|
"""
|
||
|
|
Signals module for the mailbox_manager app.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import logging
|
||
|
|
from datetime import timedelta
|
||
|
|
|
||
|
|
from django.conf import settings
|
||
|
|
from django.db.models.signals import post_save
|
||
|
|
from django.dispatch import receiver
|
||
|
|
from django.utils import timezone
|
||
|
|
|
||
|
|
from core.models import User
|
||
|
|
|
||
|
|
from mailbox_manager import enums
|
||
|
|
from mailbox_manager.models import DomainInvitation, MailDomainAccess
|
||
|
|
from mailbox_manager.utils.dimail import DimailAPIClient
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
|
||
|
|
@receiver(post_save, sender=User)
|
||
|
|
def convert_domain_invitations(sender, created, instance, **kwargs): # pylint: disable=unused-argument
|
||
|
|
"""
|
||
|
|
Convert valid domain invitations to domain accesses for a given user.
|
||
|
|
Expired invitations are ignored.
|
||
|
|
"""
|
||
|
|
logger.info("Convert domain invitations for user %s", instance)
|
||
|
|
if created:
|
||
|
|
valid_domain_invitations = DomainInvitation.objects.filter(
|
||
|
|
email=instance.email,
|
||
|
|
created_at__gte=(
|
||
|
|
timezone.now()
|
||
|
|
- timedelta(seconds=settings.INVITATION_VALIDITY_DURATION)
|
||
|
|
),
|
||
|
|
)
|
||
|
|
|
||
|
|
if not valid_domain_invitations.exists():
|
||
|
|
return
|
||
|
|
|
||
|
|
MailDomainAccess.objects.bulk_create(
|
||
|
|
[
|
||
|
|
MailDomainAccess(
|
||
|
|
user=instance, domain=invitation.domain, role=invitation.role
|
||
|
|
)
|
||
|
|
for invitation in valid_domain_invitations
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
management_role = set(valid_domain_invitations.values_list("role", flat="True"))
|
||
|
|
if (
|
||
|
|
enums.MailDomainRoleChoices.OWNER in management_role
|
||
|
|
or enums.MailDomainRoleChoices.ADMIN in management_role
|
||
|
|
):
|
||
|
|
# Sync with dimail
|
||
|
|
dimail = DimailAPIClient()
|
||
|
|
dimail.create_user(instance.sub)
|
||
|
|
|
||
|
|
for invitation in valid_domain_invitations:
|
||
|
|
if invitation.role in [
|
||
|
|
enums.MailDomainRoleChoices.OWNER,
|
||
|
|
enums.MailDomainRoleChoices.ADMIN,
|
||
|
|
]:
|
||
|
|
dimail.create_allow(instance.sub, invitation.domain.name)
|
||
|
|
|
||
|
|
valid_domain_invitations.delete()
|
||
|
|
logger.info("Invitations converted to domain accesses for user %s", instance)
|