Base Model file
from django.db import models
# Create your models here.
class BaseModel(models.Model):
created_at = models.DateTimeField(auto_now_add=True)
updated_at = models.DateTimeField(auto_now=True)
class Meta:
abstract = True
User Model file
from django.db import models
from django.contrib.auth.models import (
AbstractBaseUser,
BaseUserManager,
PermissionsMixin,
)
from core.models import BaseModel
# Create your models here.
class UserManager(BaseUserManager):
def create_user(self, email, password=None, **extra_fields):
if not email:
raise ValueError("The Email field must be set")
email = self.normalize_email(email)
user = self.model(email=email, **extra_fields)
user.set_password(password)
user.save(using=self._db)
return user
def create_superuser(self, email, password=None):
user = self.create_user(email, password)
user.is_staff = True
user.is_superuser = True
user.is_active = True
user.save(using=self._db)
return user
class User(AbstractBaseUser, BaseModel, PermissionsMixin):
email = models.EmailField(max_length=255, unique=True, db_index=True)
first_name = models.CharField(max_length=100, blank=True)
last_name = models.CharField(max_length=100, blank=True)
is_active = models.BooleanField(default=True)
is_staff = models.BooleanField(default=False)
objects = UserManager()
USERNAME_FIELD = "email"
REQUIRED_FIELDS = []
def get_full_name(self):
if self.first_name and self.last_name:
return f"{self.first_name} {self.last_name}"
else:
return None
def __str__(self):
return self.email
admin.py File
from django.contrib import admin
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from accounts.models import User
# Register your models here.
class UserAdmin(BaseUserAdmin):
ordering = ["id"]
list_display = ["email", "first_name", "last_name"]
fieldsets = (
(None, {"fields": ("email", "password")}),
("Personal Info", {"fields": ("first_name", "last_name")}),
(
"Permissions",
{"fields": ("is_active", "is_staff", "is_superuser")},
),
)
add_fieldsets = (
(None, {"classes": ("wide",), "fields": ("email", "password1", "password2")}),
)
admin.site.register(User, UserAdmin)