Skip to content

Instantly share code, notes, and snippets.

@4k45hv3rm4
Created July 15, 2020 13:30
Show Gist options
  • Save 4k45hv3rm4/21cd61bb68cbecf3a39b041f659bf9f2 to your computer and use it in GitHub Desktop.
Save 4k45hv3rm4/21cd61bb68cbecf3a39b041f659bf9f2 to your computer and use it in GitHub Desktop.
from django.db import models
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager
#manager for our custom model
class MyAccountManager(BaseUserManager):
"""
This is a manager for Account class
"""
def create_user(self, email, username, password=None):
if not email:
raise ValueError("Users must have an Emaill address")
if not username :
raise ValueError("Users must have an Username")
user = self.model(
email=self.normalize_email(email),
username=username,
)
user.set_password(password)
user.save(using=self._db)
return user
def create_superuser(self, email, username, password):
user = self.create_user(
email=self.normalize_email(email),
password=password,
username=username,
)
user.is_admin = True
user.is_staff=True
user.is_superuser=True
user.save(using=self._db)
return user
class Account(AbstractBaseUser):
"""
Custom user class inheriting AbstractBaseUser class
"""
email = models.EmailField(verbose_name='email', max_length=60, unique=True)
username = models.CharField(max_length=30,unique=True)
date_joined = models.DateTimeField(verbose_name='date joined', auto_now_add=True)
last_login = models.DateTimeField(verbose_name="last login", auto_now=True)
is_admin = models.BooleanField(default=False)
is_active = models.BooleanField(default=True)
is_staff = models.BooleanField(default=False)
is_superuser = models.BooleanField(default=False)
USERNAME_FIELD = 'email'
REQUIRED_FIELDS = ['username']
objects = MyAccountManager()
def __str__(self):
return self.email
def has_perm(self, perm, obj=None):
return self.is_admin
def has_module_perms(self, app_label ):
return True
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment