from django.contrib.auth.models import BaseUserManager from django.core.exceptions import ValidationError from django.utils.translation import gettext_lazy as _ from django.core.validators import validate_email class CustomUserManager(BaseUserManager): def valide_email(self, email): if not email: raise ValidationError(_('The Email field must be set')) try: validate_email(email) except ValidationError: raise ValidationError(_('Enter a valid email address.')) return self.normalize_email(email) def create_user(self, email, first_name, last_name, password=None, **extra_fields): email = self.valide_email(email) user = self.model(email=email, first_name=first_name, last_name=last_name, **extra_fields) user.set_password(password) user.save(using=self._db) return user def create_superuser(self, email, first_name, last_name, password=None, **extra_fields): extra_fields.setdefault('is_staff', True) extra_fields.setdefault('is_superuser', True) extra_fields.setdefault('is_active', True) extra_fields.setdefault('isVerified', 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, first_name, last_name, password, **extra_fields)