このチャプターの目次

カスタムユーザーモデル

カスタムユーザーモデルを作成していきます。

userId に uuid4 を使用してユニークな値を設定します。

Google 認証で取得した username と email をデータベースに格納します。

accounts/models.py

from django.db import models
from django.contrib.auth.models import AbstractBaseUser, PermissionsMixin, BaseUserManager
from uuid import uuid4
from django.utils import timezone


class UserManager(BaseUserManager):
    def create_user(self, username, email, password=None):
        if not email:
            raise ValueError('メールアドレスは必須です')

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

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

        return user

    def create_superuser(self, username, email, password):
        user = self.create_user(username, email, password)
        user.is_staff = True
        user.is_superuser = True
        user.save(using=self._db)

        return user


class CustomUser(AbstractBaseUser, PermissionsMixin):
    userId = models.CharField(max_length=255, default=uuid4, primary_key=True, editable=False)
    username = models.CharField('名前', max_length=255, unique=True)
    email = models.EmailField('メールアドレス', unique=True)
    created = models.DateTimeField('入会日', default=timezone.now)

    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)

    objects = UserManager()

    USERNAME_FIELD = 'username'
    REQUIRED_FIELDS = ['email']

    def __str__(self):
        return self.email

管理画面

管理画面でユーザー情報を確認できるように設定します。

accounts/admin.py

from django.contrib import admin
from .models import CustomUser

admin.site.register(CustomUser)