"""User Manager used by Improved User; may be extended"""
from django.contrib.auth.models import BaseUserManager
[docs]class UserManager(BaseUserManager):
"""Manager for Users; overrides create commands for new fields
Meant to be interacted with via the user model.
.. code:: python
User.objects # the UserManager
User.objects.all() # has normal Manager/UserManager methods
User.objects.create_user # overrides methods for Improved User
Set to :attr:`~django.db.models.Model.objects` by
:attr:`~improved_user.models.AbstractUser`
"""
def _create_user(
self, email, password, is_staff, is_superuser, **extra_fields):
"""Helper method to save a User with improved user fields"""
if not email:
raise ValueError('An email address must be provided.')
if 'username' in extra_fields:
raise ValueError(
'The Improved User model does not have a username; '
'it uses only email')
user = self.model(
email=self.normalize_email(email),
is_staff=is_staff, is_superuser=is_superuser,
**extra_fields)
user.set_password(password)
user.save(using=self._db)
return user
[docs] def create_user(self, email=None, password=None, **extra_fields):
"""Save new User with email and password"""
extra_fields.setdefault('is_staff', False)
extra_fields.setdefault('is_superuser', False)
return self._create_user(email, password, **extra_fields)
[docs] def create_superuser(self, email, password, **extra_fields):
"""Save new User with is_staff and is_superuser set to True"""
extra_fields.setdefault('is_staff', True)
extra_fields.setdefault('is_superuser', True)
if extra_fields.get('is_staff') is not True:
raise ValueError('Superuser must have is_staff=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)