Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class UserManager(BaseUserManager):
- use_in_migrations = True
- def _create_user(self, email, password, **extra_fields):
- if not email:
- raise ValueError('The given email must be set')
- email = self.normalize_email(email)
- user = self.model(email=email, **extra_fields)
- user.set_password(password)
- user.save(using=self._db)
- return user
- def create_user(self, email, password=None, **extra_fields):
- extra_fields.setdefault('is_superuser', False)
- return self._create_user(email, password, **extra_fields)
- def create_superuser(self, email, password, **extra_fields):
- extra_fields.setdefault('is_superuser', True)
- if extra_fields.get('is_superuser') is not True:
- raise ValueError('Superuser must have is_superuser=True.')
- return self._create_user(email, password, **extra_fields)
- class UserProfile(models.Model):
- avatar = models.URLField()
- user = models.ForeignKey(User, unique=True)
- objects = UserManager()
- @receiver(post_save, sender=User)
- def create_user_profile(sender, **kwargs):
- #When creating a new user, make a profile for him or her.
- u = kwargs["instance"]
- if not UserProfile.objects.filter
- UserProfile(user=u).save()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement