本文将帮助大家在Django中创建自己的自定义用户模型
自定义用户模型有什么需求?
创建自己的自定义用户模型至关重要。将来,如果要对模型进行一些更改,则可以轻松进行这些更改。不然我们可能必须对模型进行一些更改,而且代码的某些部分也将被更改。在某些网站中,与用户名相比,使用电子邮件进行登录是有意义的。因此,对于程序员而言,拥有自定义用户模型是一个好习惯。
本文将帮助大家根据需要创建自己的自定义用户模型。请继续执行以下步骤。
- Django 安装
- 创建Django应用
- 该应用程序的自定义模型
- 迁移数据库
1. Django安装
设置Django之后,进入创建的环境。然后,我们可以通过以下命令创建Django项目。
- django-admin startproject mysite
2.创建Django应用
如果发现某些网站使用了类似/ accounts /的URL,它们看起来就是专业代码。因此,我们还需要创建一个名为accounts的应用程序,在其中管理与该帐户有关的所有事情。因此,在当前环境中,使用以下命令创建Django应用。
- python manage.py startapp accounts
创建后,该应用将包含基本文件。创建的文件如下所示。
- accounts/
- __init__.py
- admin.py
- apps.py
- models.py
- tests.py
- views.py
一旦创建,所有的数据库模型代码都应该写在models.py文件中。另外,不要忘记创建一个urls.py文件,该文件有助于通过帐户应用程序路由链接。创建完成后,让我们进入根据我们的需求构建自定义用户模型的下一步。
3.该应用程序的自定义模型
因此,为了创建自定义用户模型,我们需要在models.py文件中写下以下代码,因为python使用对象来管理和访问存储在数据库中的数据。下面提到了models.py文件的代码。
- from django.db import models
- from django.contrib.auth.models import AbstractBaseUser , BaseUserManager
- import uuid
- class UserManager(BaseUserManager):
- def create_user(self, email, firstname, lastname, phone,password=None):
- user = self.model(
- email = self.normalize_email(email),
- firstnamefirstname = firstname,
- lastnamelastname = lastname,
- phonephone = phone,
- )
- user.set_password(password)
- user.save(using = self._db)
- return user
- def create_superuser(self, email,firstname,lastname,phone,password=None):
- user = self.create_user(
- emailemail=email,
- passwordpassword=password,
- firstnamefirstname = firstname,
- lastnamelastname = lastname,
- phonephone = phone,
- )
- user.is_admin = True
- user.is_staff = True
- user.save(using=self._db)
- return user
- class User(AbstractBaseUser):
- id = models.CharField(max_length=200, default=uuid.uuid4,unique=True,primary_key=True)
- email = models.EmailField(null=False, max_length=100,unique=True)
- firstname = models.CharField(null=False, max_length=100)
- lastname = models.CharField(null=False, max_length=100)
- phone = models.IntegerField(null=False,unique=True)
- date_joined = models.DateTimeField(auto_now=True)
- last_login = models.DateTimeField(auto_now=True)
- is_admin = models.BooleanField(default = False)
- is_active = models.BooleanField(default = True)
- is_staff = models.BooleanField(default = False)
- is_superuser = models.BooleanField(default = False)
- USERNAME_FIELD = 'email'
- REQUIRED_FIELDS = ['firstname','lastname','phone']
- objects = UserManager()
- def __str__(self):
- return self.email + ", " + self.firstname
- def has_perm(self, perm, obj = None):
- return self.is_admin
- def has_module_perms(self, app_label):
- return True
用户类包含表的字段或结构,我们在其中获取有关用户的一些详细信息以及其中的某些条件。为了管理User类,我们有UserManager,它将用于创建新用户或超级用户。就像我之前说的,这里的用户名字段将被替换为电子邮件。它比用户名字段更有意义。还提到了必填字段。通过使用电子邮件作为用户名字段,它会自动将电子邮件字段添加为必不可少的字段。
因此,通过UserManager,我们可以创建一个新的普通用户和一个超级用户。user.set_password()会将纯文本密码哈希为默认的哈希算法。Django将PBKDF2与SHA-256哈希一起使用,并具有密码扩展机制。user.save()函数将使用字段和哈希密码保存创建的用户。创建自定义用户模型很简单。
在这里需要注意,在创建模型之前不要迁移模型。在此之前,创建模型后,我们必须使用自定义模型更新Django设置文件,否则Django将使用其自己的预先实现的模型。只需在项目目录中的settings.py文件中使用此代码即可。
- AUTH_USER_MODEL = ‘accounts.User’
设置好模型后,就该将模型迁移到数据库了。
4.迁移模型
现在,我们准备通过将更改迁移到数据库中来迁移自定义用户模型。使用以下命令在当前环境中迁移并执行该代码。
- python manage.py makemigrations
- python manage.py migrate
此命令将查找Django项目中的所有模型和模型中的更改。如果要专门为应用程序进行迁移,可以使用以下命令进行。
- python manage.py makemigrations <app_name>
- python manage.py migrate <app_name>
可以将<app_name>替换为帐户,以仅在该应用程序上进行更改。