如何在Django中創建自己的自定義用戶模型?

本文將幫助大家在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),
            firstname = firstname,
            lastname = lastname,
            phone = 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(
            email=email,
            password=password,
            firstname = firstname,
            lastname = lastname,
            phone = 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>替換爲帳戶,以僅在該應用程序上進行更改。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章