mygpo/users/apps.py

   1 from django.apps import AppConfig, apps
   2 from django.contrib.auth import get_user_model
   3 from django.db.models.signals import post_save
   4
   5
   6 def create_missing_profile(sender, **kwargs):
   7     """ Creates a UserProfile if a User doesn't have one """
   8     user = kwargs['instance']
   9
  10     if not hasattr(user, 'profile'):
  11         UserProfile = apps.get_model('users.UserProfile')
  12         profile = UserProfile.objects.create(user=user)
  13         user.profile = profile
  14
  15
  16 class UsersConfig(AppConfig):
  17     name = 'mygpo.users'
  18     verbose_name = "Users and Clients"
  19
  20     def ready(self):
  21         User = get_user_model()
  22         post_save.connect(create_missing_profile, sender=User)