Advertisement
Guest User

Untitled

a guest
Sep 2nd, 2017
603
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.36 KB | None | 0 0
  1. class CustomManager(UserManager):
  2. use_in_migrations = True
  3.  
  4. def _create_user(self, username, email, password, age, **extra_fields):
  5. """
  6. Creates and saves a User with the given username, email and password.
  7. """
  8. if not username:
  9. raise ValueError('The given username must be set')
  10. email = self.normalize_email(email)
  11. username = self.model.normalize_username(username)
  12. user = self.model(username=username, email=email, **extra_fields)
  13. user.set_password(password)
  14. user.save(using=self._db)
  15. return user
  16.  
  17. def create_user(self, username, age, email=None, password=None, **extra_fields):
  18. extra_fields.setdefault('is_staff', False)
  19. extra_fields.setdefault('is_superuser', False)
  20. return self._create_user(username,age, email, password, **extra_fields)
  21.  
  22. class Profile(AbstractUser):
  23. #user = models.OneToOneField(User, on_delete=models.CASCADE)
  24. objects = CustomManager()
  25. Family = models.CharField(max_length=200)
  26. City = models.CharField(max_length=200)
  27. GENDER_CHOICES = (
  28. ('M', 'Male'),
  29. ('F', 'Female'),
  30. )
  31. sex = models.CharField(max_length=1, choices=GENDER_CHOICES) #models.BooleanField(default=True)
  32. Age = models.SmallIntegerField(null=True)
  33. photo = models.ImageField(upload_to="images")
  34.  
  35. profile = Profile.objects.create_user(i_name,e_mail,pswd,age)
  36.  
  37. user = self.model(username=username, email=email, **extra_fields)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement