User 모델 확장

User 모델 확장이란?

기존 Django에 내장된 유저에 추가 필드들(성별 등)을 붙여 커스텀마이징 하는 것

 

- 확장 방식의 종류

1) proxy model 사용하기

2) 하나의 모델을 정의 후, User 모델과 One-to-One 관계 형성

3) AbstractBaseUser

4) AbstractUser

 

- AbstractUser 사용 예

 

from django.db import models
from django.contrib.auth.models import AbstractUser

class CustomUser(AbstractUser):
	... # 추가 정의

 

- 실제 사용 예

1) users App 생성

$ python manage.py startapp users

2) settings.py 내에 INSTALLED_APP 추가 및 AUTH_USER_MODEL 추가

# manage.py

INSTALLED_APPS = [
    'django.contrib.admin',
    'django.contrib.auth',
    'django.contrib.contenttypes',
    'django.contrib.sessions',
    'django.contrib.messages',
    'django.contrib.staticfiles',
    'users', # 추가
]

AUTH_USER_MODEL = 'users.User' # <app명.class명>

3) models.py 내에 유저 클래스 생성

from django.db import models
from django.contrib.auth.models import AbstractUser, BaseUserManager

GENDER_CHOICES = (

    (0, 'Male'),

    (1, 'Female'),

    (2, 'Not to disclose')

)

# AbstractUser 구현시, 사용하는 헬퍼(Helper) 클래스
# 주요 메소드 : create_user, create_superuser
class UserManager(BaseUserManager):

    def _create_user(self, email, username, password, gender=2, **extra_fields):

        """

        Create and save a user with the given username, email, and password.

        """

        if not email:
            raise ValueError('The given email must be set')

        email = self.normalize_email(email)

        username = self.model.normalize_username(username)

        user = self.model(email=email, username=username, gender=gender, **extra_fields)

        user.set_password(password)

        user.save(using=self._db)

        return user

    def create_user(self, email, username='', password=None, **extra_fields):

        extra_fields.setdefault('is_staff', False)

        extra_fields.setdefault('is_superuser', False)

        return self._create_user(email, username, password, **extra_fields)

    def create_superuser(self, email, password, **extra_fields):

        extra_fields.setdefault('is_staff', True)

        extra_fields.setdefault('is_superuser', True)

        if extra_fields.get('is_staff') is not True:
            raise ValueError('Superuser must have is_staff=True.')

        if extra_fields.get('is_superuser') is not True:
            raise ValueError('Superuser must have is_superuser=True.')

        return self._create_user(email, 'blogs/like_section.html', password, **extra_fields)


class User(AbstractUser):
    email = models.EmailField(verbose_name='email', max_length=255, unique=True)

    username = models.CharField(max_length=30)

    gender = models.SmallIntegerField(choices=GENDER_CHOICES)

    objects = UserManager()

    USERNAME_FIELD = 'email'

    REQUIRED_FIELDS = []  # 필수로 받고 싶은 필드들 넣기 원래 소스 코드엔 email필드가 들어가지만 우리는 로그인을 이메일로 하니깐..

    def __str__(self):
        return "<%d %s>" % (self.pk, self.email)

4) DB 변경사항 기록 및 생성

$ python manage.py makemigrations users
$ python manage.py migrate

5) 관리자에 등록

# users/admin.py
from django.contrib import admin

from .models import User

# Register your models here.
admin.site.register(User) # 등록