Advertisement
BlueBear

userprofile_issue

Mar 3rd, 2017
224
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.15 KB | None | 0 0
  1. class UserManager(BaseUserManager):
  2.     use_in_migrations = True
  3.     def _create_user(self, email, password, **extra_fields):
  4.  
  5.     if not email:
  6.         raise ValueError('The given email must be set')
  7.     email = self.normalize_email(email)
  8.     user = self.model(email=email, **extra_fields)
  9.     user.set_password(password)
  10.     user.save(using=self._db)
  11.     return user
  12.     def create_user(self, email, password=None, **extra_fields):
  13.         extra_fields.setdefault('is_superuser', False)
  14.         return self._create_user(email, password, **extra_fields)
  15.     def create_superuser(self, email, password, **extra_fields):
  16.     extra_fields.setdefault('is_superuser', True)
  17.     if extra_fields.get('is_superuser') is not True:
  18.         raise ValueError('Superuser must have is_superuser=True.')
  19.     return self._create_user(email, password, **extra_fields)
  20. class UserProfile(models.Model):
  21.     avatar = models.URLField()
  22.     user = models.ForeignKey(User, unique=True)
  23.     objects = UserManager()
  24.     @receiver(post_save, sender=User)
  25.     def create_user_profile(sender, **kwargs):
  26.         #When creating a new user, make a profile for him or her.
  27.         u = kwargs["instance"]
  28.         if not UserProfile.objects.filter
  29.         UserProfile(user=u).save()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement