在 django_Restframework 中使用 AbstractUser 创建自定义用户模型

每个新的 Django 项目都应该使用自定义用户模型。 Django 官方文档说它是“强烈推荐的”，但我会更进一步，毫不犹豫地说：你不预先使用自定义用户模型是疯了。

为什么你需要在 Django 中自定义用户模型？

当您使用自定义用户模型开始您的项目时，请停止考虑这是否是您项目的正确选择。

将所有与用户相关的信息保存在一个模型中，无需额外或更复杂的数据库查询来检索相关模型。另一方面，将特定于应用程序的用户信息存储在与您的自定义用户模型有关系的模型中可能更合适。这允许每个应用程序指定自己的用户数据要求，而不会与其他应用程序的假设发生潜在冲突或破坏。这也意味着您将保持您的用户模型尽可能简单，专注于身份验证，并遵循 Django 期望自定义用户模型满足的最低要求。

所以我认为很清楚为什么我们需要在 django 中自定义用户模型，在本文中我们将学习如何在 django 中创建自定义用户模型及其 api

创建自定义用户模型的步骤

为我们的代码创建并导航到名为 users 的专用目录

$ cd ~/Desktop$ mkdir code && cd code

安装 Django

$ pipenv install django

新建一个名为 login 的 Django 项目

$ django-admin startproject login

制作一个新的应用程序 API 并安装 rest 框架

$ python manage.py startapp api
$ pipenv install rest_framework

现在我们需要配置我们的设置。 py如下

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',
  ),
}

在此之后，我们需要创建一个自定义用户模型。为此，请更改您的 models.py 文件。在这里，我们扩展 AbstractUser 并将身份验证凭据更改为电子邮件，我们还在自定义用户中添加了一些额外的字段

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)

在此之后，我们还需要将这些更改保存到管理面板。将此代码添加到admin.py

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)

现在在您的应用程序中创建一个 serializers.py 文件。将以下代码添加到您的serializers.py

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

我们快完成了。现在在你的应用程序中创建 api/urls.py 文件并添加以下几行

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

现在将以下代码添加到项目的 login/urls.py

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

我们都设置了自定义用户模型。现在将这些更改保存到您的项目中并通过 CLI 进行迁移

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

现在创建一个超级用户

$ python manage.py createsuperuser

Email address: test@test.com

Password:

Password (again):

Superuser created successfully.

现在运行服务器。

$ python manage.py runserver