from django.db.models.signals import post_save
from django.dispatch import receiver
from django.contrib.auth.models import User

from .models import UserProfile


@receiver(post_save, sender=User)
def create_user_profile(sender, instance, created, **kwargs):
    if created:
        user_type = "administrator" if instance.is_staff else "user"
        UserProfile.objects.create(user=instance, user_type=user_type)