Advertisement
Guest User

Untitled

a guest
Aug 16th, 2016
77
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.76 KB | None | 0 0
  1. class MyUser(AbstractBaseUser, PermissionsMixin):
  2. USER_ACCOUNT_TYPE_CHOICES = (
  3. (1, _('organization')),
  4. (2, _('personal')),
  5. )
  6.  
  7. user_account_type = models.IntegerField(_('account type'), choices=USER_ACCOUNT_TYPE_CHOICES)
  8. email = models.EmailField(_('email address'), max_length=255, unique=True)
  9.  
  10. class MyUserManager(BaseUserManager):
  11. def _create_user(self, email, password, username=None, **extra_fields):
  12. """
  13. Creates and saves a User with the given username, email and password.
  14. """
  15. if not email:
  16. raise ValueError('The given email must be set')
  17. email = self.normalize_email(email)
  18. if username:
  19. username = self.model.normalize_username(username)
  20. user = self.model(username=username, email=email, **extra_fields)
  21. else:
  22. user = self.model(email=email, **extra_fields)
  23.  
  24. user.set_password(password)
  25. user.save(using=self._db)
  26. return user
  27.  
  28. def create_user(self, email=None, password=None, username=None, **extra_fields):
  29. extra_fields.setdefault('is_staff', False)
  30. extra_fields.setdefault('is_superuser', False)
  31. return self._create_user(email, password, username=username, **extra_fields)
  32.  
  33. def create_superuser(self, email, password, username=None, **extra_fields):
  34. extra_fields.setdefault('is_staff', True)
  35. extra_fields.setdefault('is_superuser', True)
  36.  
  37. if extra_fields.get('is_staff') is not True:
  38. raise ValueError('Superuser must have is_staff=True.')
  39. if extra_fields.get('is_superuser') is not True:
  40. raise ValueError('Superuser must have is_superuser=True.')
  41.  
  42. return self._create_user(email, password, username=username, **extra_fields)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement