在Django中创建自定义密码验证的方法是什么？

2 年 ago

清, 扬

3 minutes

简述

我想解释如何在Django内创建自定义验证来处理用户的密码等问题。

文件结构

本次要创建的文件结构如下所示。

tree
・
└── application
    ├── models.py
    ├── utils
    │   ├── __init__.py
    │   └── password_validator.py
    └── project
        └── settings.py

models.py(カスタムユーザの作成)

に設定を記載していきます

password_validator.py

re.fullmatch()関数を使って指定した正規表現と完全に一致するかどうかを調べます

import re

from rest_framework.exceptions import ValidationError


class PasswordValidator:
    def validate(self, password, user=None):
        REX = r"(?=.*\d)(?=.*[a-z])(?=.*[A-Z])(?=.*[~!@#$%^&*_\-+=`|(){}\[\]:;\"\'<>,.?/]).{8,64}"
        result = re.fullmatch(REX, password)
        if not result:
            raise ValidationError("英大小文字、数字、または特殊文字が含まれていません")

正規表現の書き方

正規表現の書き方は以下の通りです

正規表現説明?=.*一致するかどうか（\d）少なくとも1つの数字（[a-z]）少なくとも1つの小文字のアルファベット（[A-Z]）少なくとも1つの大文字のアルファベット*[~!@#$%^&*_-+=`|(){}[]:;”‘<>,.?/]少なくとも1つの特殊文字{8,64}全体で8文字以上、64文字以下（.{8,64}）

settings.py

AUTH_PASSWORD_VALIDATORSに該当するクラス名もしくはファイル名を記載します

AUTH_PASSWORD_VALIDATORS = [
    {
        "NAME": "django.contrib.auth.password_validation.UserAttributeSimilarityValidator",
    },
    {
        "NAME": "django.contrib.auth.password_validation.MinimumLengthValidator",
    },
    {
        "NAME": "django.contrib.auth.password_validation.CommonPasswordValidator",
    },
    {
        "NAME": "django.contrib.auth.password_validation.NumericPasswordValidator",
    },
    # 以下にカスタムバリデーションのクラス名を記載
    {
        "NAME": "application.utils.password_validator.PasswordValidator",
    },
]

models.py

システムユーザ用のModelを作成します
詳細については以下の記事を参照してください

import uuid

from django.contrib.auth.models import AbstractUser
from django.contrib.auth.validators import UnicodeUsernameValidator
from django.core.validators import RegexValidator
from django.db import models


class User(AbstractUser):
    """システムユーザ"""

    username_validator = UnicodeUsernameValidator()

    class Role(models.IntegerChoices):
        """システムユーザのロール

        Args:
            MANAGEMENT(0): 管理者
            GENERAL(1):    一般
            PART_TIME(2):  アルバイト
        """

        MANAGEMENT = 0
        GENERAL = 1
        PART_TIME = 2

    # 不要なフィールドはNoneにすることができる
    first_name = None
    last_name = None
    date_joined = None
    groups = None
    id = models.UUIDField(
        primary_key=True,
        default=uuid.uuid4,
        editable=False,
        db_comment="システムユーザID",
    )
    employee_number = models.CharField(
        unique=True,
        validators=[RegexValidator(r"^[0-9]{8}$")],
        max_length=8,
        # 管理者のログイン画面で社員番号と表示される
        verbose_name="社員番号",
        db_comment="社員番号",
    )
    username = models.CharField(
        max_length=150,
        unique=True,
        validators=[username_validator],
        db_comment="ユーザ名",
    )
    email = models.EmailField(
        max_length=254,
        unique=True,
        db_comment="メールアドレス",
    )
    role = models.PositiveIntegerField(
        choices=Role.choices,
        default=Role.PART_TIME,
        db_comment="システムユーザのロール",
    )
    created_at = models.DateTimeField(
        auto_now_add=True,
        db_comment="作成日",
    )
    updated_at = models.DateTimeField(
        auto_now=True,
        db_comment="更新日",
    )

    USERNAME_FIELD = "employee_number"
    REQUIRED_FIELDS = ["email", "username"]

    class Meta:
        ordering = ["employee_number"]
        db_table = "User"
        db_table_comment = "システムユーザ"

    def __str__(self):
        return self.username

実際に検証してみよう！

python manage.py createsuperuser

を実行します

以下のように入力していきます
今回は数字だけのパスワードを入力します

社員番号: 00000001
Email: test@gmail.com
Password: 11111111
Password (again): 11111111

パスワード内に英大小文字、特殊文字が含まれていないため、以下のようにValidationErrorが表示されたら成功です

File "/code/crm/utils/crm_validator.py", line 51, in validate
    raise ValidationError("英大小文字、数字、または特殊文字が含まれていません")
rest_framework.exceptions.ValidationError: [ErrorDetail(string='英大小文字、数字、または特殊文字が含まれていません', code='invalid')]

他のパターンも気になる方は以下のようにテストコードで検証してみましょう

import pytest

from crm.utils.crm_validator import PasswordValidator
from rest_framework.exceptions import ValidationError

@pytest.mark.django_db()
class TestPasswordValidator:
    def test_user_password_does_not_contain_non_alphanumeric_characters(self):
        """異常/パスワードに不英数文字が含まれていない"""
        with pytest.raises(ValidationError):
            PasswordValidator.validate(self, password="TestingUsr01")

    def test_user_password_does_not_contain_numbers(self):
        """異常/パスワードに数字が含まれていない"""
        with pytest.raises(ValidationError):
            PasswordValidator.validate(self, password="T@stingUsr")

    def test_user_password_does_not_contain_uppercase_letters(self):
        """異常/パスワードに大文字が含まれていない"""
        with pytest.raises(ValidationError):
            PasswordValidator.validate(self, password="t@stingusr01")

    def test_user_password_is_less_than_7_characters(self):
        """異常/パスワードが7文字以下"""
        with pytest.raises(ValidationError):
            PasswordValidator.validate(self, password="T@stus1")

    def test_user_password_is_65_characters_or_longer(self):
        """異常/パスワードが65文字以上"""
        with pytest.raises(ValidationError):
            PasswordValidator.validate(self, password="a" * 65)

简述