Related Articles

Related Articles

Creating custom user model API extending AbstractUser in Django
  • Last Updated : 23 Oct, 2020

Every new Django project should use a custom user model. The official Django documentation says it is “highly recommended” but I’ll go a step further and say without hesitation: You are straight up crazy not to use a custom user model up front.

Why do u need a custom user model in Django? 

When you start your project with a custom user model, stop to consider if this is the right choice for your project. Keeping all user related information in one model removes the need for additional or more complex database queries to retrieve related models. On the other hand, it may be more suitable to store app-specific user information in a model that has a relation with your custom user model. That allows each app to specify its own user data requirements without potentially conflicting or breaking assumptions by other apps. It also means that you would keep your user model as simple as possible, focused on authentication, and following the minimum requirements Django expects custom user models to meet.

So i think its clear why we need a custom user model in django, here in this article we are going to learn how to create custom user model and its api in django now, lets begin with the coding part . 

Setup –



  • Create and navigate into a dedicated directory called users for our code

  • make a new Django project called login

  • make a new app users

  • start the local web server

Commands –

$ cd ~/Desktop
$ mkdir code && cd code $ pipenv install django $ pipenv shell $ django-admin startproject login $ python manage.py startapp api $ pipenv install rest_framework

Now, add following code in settings.py,

filter_none

edit
close

play_arrow

link
brightness_4
code

INSTALLED_APPS =
    'django.contrib.admin',
   'django.contrib.auth',
   'django.contrib.contenttypes',
   'django.contrib.sessions',
   'django.contrib.messages',
   'django.contrib.staticfiles',
   # Add these lines to to your 
   # installed apps section in settings. py
   'rest_framework',
   'rest_framework.authtoken',
   'api',
   'rest_auth'
]
AUTH_USER_MODEL ='api.urls'
REST_FRAMEWORK = {
   'DEFAULT_PERMISSION_CLASSES': (
       'rest_framework.permissions.IsAuthenticated',
   ),
   'DEFAULT_AUTHENTICATION_CLASSES': (
       'rest_framework_jwt.authentication.JSONWebTokenAuthentication',
       'rest_framework.authentication.SessionAuthentication',
       'rest_framework.authentication.BasicAuthentication',
   ),
}

chevron_right


After this we need to create a custom user model. For this change your models.py file as following. Here we are extending AbstractUser and changing authentication credentials to Email. And we are also adding some extra fields in our custom user

filter_none

edit
close

play_arrow

link
brightness_4
code

from django.db import models
from django.contrib.auth.models import AbstractUser
from django.utils.translation import ugettext_lazy as _
from django.conf import settings
from datetime import date
class User(AbstractUser):
   username = models.CharField(max_length = 50, blank = True, null = True, unique = True)
   email = models.EmailField(_('email address'), unique = True)
   native_name = models.CharField(max_length = 5)
   phone_no = models.CharField(max_length = 10)
   USERNAME_FIELD = 'email'
   REQUIRED_FIELDS = ['username', 'first_name', 'last_name']
   def __str__(self):
       return "{}".format(self.email)

chevron_right


After this we need to save these changes to the admin panel as well. Add this code to the admin.py



filter_none

edit
close

play_arrow

link
brightness_4
code

from django.contrib import admin
from django.utils.translation import ugettext_lazy as _
from django.contrib.auth.admin import UserAdmin as BaseUserAdmin
from django.contrib.auth import get_user_model
from django.contrib.auth.admin import UserAdmin
from .models import User
class UserAdmin(BaseUserAdmin):
   form = UserChangeForm
   fieldsets = (
       (None, {'fields': ('email', 'password', )}),
       (_('Personal info'), {'fields': ('first_name', 'last_name')}),
       (_('Permissions'), {'fields': ('is_active', 'is_staff', 'is_superuser',
                                      'groups', 'user_permissions')}),
       (_('Important dates'), {'fields': ('last_login', 'date_joined')}),
         (_('user_info'), {'fields': ('native_name', 'phone_no')}),
   )
   add_fieldsets = (
       (None, {
           'classes': ('wide', ),
           'fields': ('email', 'password1', 'password2'),
       }),
   )
   list_display = ['email', 'first_name', 'last_name', 'is_staff', 'native_name', 'phone_no']
   search_fields = ('email', 'first_name', 'last_name')
   ordering = ('email', )
admin.site.register(User, UserAdmin)

chevron_right


Now create a serializers.py file in your app. Add following code to your serializers. py

filter_none

edit
close

play_arrow

link
brightness_4
code

rom rest_framework import serializers
from api.models import  User
class UserSerializer(serializers.ModelSerializer):
   class Meta:
       model = User
       fields = "__all__"

chevron_right


Now create api/urls.py  file inside you app and add following lines

filter_none

edit
close

play_arrow

link
brightness_4
code

from django.contrib import admin
from django.urls import path
from django.conf.urls import include
urlpatterns = [
   path('auth/', include('rest_auth.urls')),
]

chevron_right


Now add following code to the login/urls.py of your project

filter_none

edit
close

play_arrow

link
brightness_4
code

from django.contrib import admin
from django.urls import path
  
urlpatterns = [
   path('admin/', admin.site.urls),
   path(" ", include("api.urls"))
]

chevron_right


We are all set with the custom user model. Now save these changes to you project and makemigrations through CLI

$ python manage.py makemigrations users
$ python manage.py migrate

Now create a superuser

$ python manage.py createsuperuser
Email address: test@test.com
Password:
Password (again):
Superuser created successfully.

Now run server with.

$ python manage.py runserver

And go through the URL. You can check your custom user in the admin pannel

Output –

Attention geek! Strengthen your foundations with the Python Programming Foundation Course and learn the basics.

To begin with, your interview preparations Enhance your Data Structures concepts with the Python DS Course.

My Personal Notes arrow_drop_up
Recommended Articles
Page :