from django.conf import settings
from django.db.models.signals import post_migrate, post_save
from django.dispatch import receiver

from .models import UserProfile
from .roles import ensure_bootstrap_role_assignments, ensure_role_groups


@receiver(post_migrate)
def workflows_post_migrate(sender, **kwargs):
    if getattr(sender, 'name', '') != 'workflows':
        return
    ensure_role_groups()
    ensure_bootstrap_role_assignments()


@receiver(post_save, sender=settings.AUTH_USER_MODEL)
def ensure_user_profile(sender, instance, created, **kwargs):
    if created:
        UserProfile.objects.get_or_create(user=instance)