首页 | 编程 | python

Django3中的自定义用户模型实例详解

2020年08月23日 作者：mindg.cn原文

介绍

嗨，在这篇文章中，我们将学习在Django 3中创建自定义用户模型，并且还将更改Django Admin的默认登录功能。 我们将使用电子邮件和密码登录。

动机

我必须为我的应用程序创建一个“自定义用户”，我能够创建模型，但问题是createsuperuser命令无法正常工作。 为了调试它，我不得不做大量研究，问题是当时大多数资源已经过时，所以我决定写这篇文章。

我做了一个GitHub仓库，所以如果您愿意，可以直接使用它（这里有说明）

让我们开始吧

首先，创建一个Django Project并创建一个名为users的应用程序

现在我们可以开始在用户应用程序中编辑models.py文件了

在进行编辑之前，先做一些理论上的准备。

Django中的经理是什么？

Manager是一个接口，通过该接口可以将数据库查询操作提供给Django模型。 Django应用程序中的每个模型至少都有一个Manager。-Django Docs

简单地说，经理为我们提供了一种管理模型的方法。 我们可以通过使模型成为Manager类的子类来实现此目的。 管理器类是可以编辑诸如createsuperuser之类的命令的地方。

现在打开models.py并将以下代码放入其中

from django.db import models
from django.contrib.auth.models import AbstractBaseUser, BaseUserManager, PermissionsMixin


class UserManager(BaseUserManager):
 def create_user(self, email, password=None):
  """
  Creates and saves a User with the given email and password.
  """
  if not email:
   raise ValueError('Users must have an email address')

  user = self.model(
   email=self.normalize_email(email),
  )

  user.set_password(password)
  user.save(using=self._db)
  return user

 def create_superuser(self, email, password):
  """
  Creates and saves a superuser with the given email and password.
  """
  user = self.create_user(
   email,
   password=password,
  )
  user.is_staff = True
  user.is_superuser = True
  user.save(using=self._db)
  return user


class CustomUser(AbstractBaseUser, PermissionsMixin):
 name = models.CharField(max_length=50, default='Anonymous')
 email = models.EmailField(max_length=100, unique=True)

 username = None

 USERNAME_FIELD = 'email'
 REQUIRED_FIELDS = []

 session_token = models.CharField(max_length=10, default=0)

 active = models.BooleanField(default=True)
 # a admin user; non super-user
 is_staff = models.BooleanField(default=False)
 is_superuser = models.BooleanField(default=False) # a superuser

 created_at = models.DateTimeField(
  auto_now_add=True, blank=True, null=True)
 updated_at = models.DateTimeField(auto_now=True)

 objects = UserManager()

这里要注意的最重要的事情是is_staff和is_superuser属性。 在其中输入错误可能会导致调试困难。

我们在这里做了什么？

我们已经为我们的用户模型进行了管理。 在其中，我们制作了两个函数，分别称为create_user和create_superuser。
顾名思义，create_user将创建一个新用户，并通过将is_staff和is_superuser设置为true来使用create_superuser创建一个超级用户。

经理之后，我们便有了通常的模式。

我们将用户名设置为none，因为我们不想包含用户名。

其中的USERNAME_FIELD表示我们声明为“电子邮件”。 这应该是唯一的。

session_token是一个可选字段。 我在那里，因为我正在制作我的自定义令牌。

models.py的最后一行指示CustomUser是UserManager的对象。

重要的事情

制作完模型后，打开settings.py文件并在其中添加一行

AUTH_USER_MODEL = 'users.CustomUser'

Django允许您通过为AUTH_USER_MODEL设置提供一个引用自定义模型的值来覆盖默认用户模型。 这条虚线对描述了Django应用程序的名称（必须在您的INSTALLED_APPS中），以及您希望用作用户模型的Django模型的名称。 -Django文档

最后步骤

现在，您可以运行迁移命令并创建超级用户。

py manage.py makemigrations
py manage.py migrate
py manage.py createsuperuser

它将要求您提供电子邮件和密码。 详细说明。

不要忘记在管理员中注册该应用

admin.site.register(CustomUser)

现在，您可以运行服务器并在管理面板中使用您的电子邮件和密码登录。

如果您有任何建议，请告诉我。

git:https://github.com/theshubhagrwl/django_custom_user_app

总结

到此这篇关于Django3中的自定义用户模型的文章就介绍到这了,更多相关Django3自定义用户模型内容请搜索来客网以前的文章或继续浏览下面的相关文章希望大家以后多多支持来客网！

相关链接