Advertisement
robertvari

User manager example

Apr 13th, 2020
343
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.91 KB | None | 0 0
  1. from django.contrib.auth.models import BaseUserManager
  2.  
  3.  
  4. class CreativeUserManager(BaseUserManager):
  5.     def create_user(self, email, full_name, password=None, superuser=False):
  6.         if not email:
  7.             raise ValueError("User must have an email!")
  8.  
  9.         if not full_name:
  10.             raise ValueError("User must have Full name!")
  11.  
  12.         user = self.model(
  13.             email=self.normalize_email(email),
  14.             full_name=full_name
  15.         )
  16.  
  17.         user.set_password(password)
  18.  
  19.         if superuser:
  20.             user.is_admin = True
  21.             user.is_staff = True
  22.             user.is_superuser = True
  23.  
  24.         user.save(using=self._db)
  25.  
  26.         return user
  27.  
  28.     def create_superuser(self, email, full_name, password):
  29.         user = self.create_user(
  30.             email,
  31.             full_name,
  32.             password,
  33.             superuser=True
  34.         )
  35.  
  36.         return user
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement