File: //home/arjun/projects/unlimited-leads/Unlimited-Leads-Be/authorization/managers.py
import uuid
from django.contrib.auth.base_user import BaseUserManager
class UnlimitedLeadUserManager(BaseUserManager):
def create_user(self, email, password, **extra_fields):
if not email:
raise ValueError("Users must have an email address")
email = self.normalize_email(email)
user = self.model(email=email, username=str(uuid.uuid4()), **extra_fields)
user.set_password(password)
user.save()
return user
def create_superuser(self, email, password, **extra_fields):
extra_fields.setdefault("is_staff", True)
extra_fields.setdefault("is_superuser", True)
extra_fields.setdefault("is_active", True)
extra_fields.setdefault("is_verified", 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)