(编辑:jimmy 日期: 2024/11/16 浏览:2)
介绍
嗨,在这篇文章中,我们将学习在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
总结