[django 블로그] User 사용자 정보 ( models.py 모델링, 커스텀 유저 모델, BaseUserManager, AbstractUser, settings.py, admin.py)

커스텀 유저 모델 (Custom User Model)

- 지금까지 사용해온 기본 유저 모델은 username을 id 로 사용한다. 하지만 email을 id로 하고싶거나 생년월일, 성별, 프로필 사진 등 다양하게 추가하고 싶을 수 있다. 이런 경우에 커스텀 유저 모델을 사용한다.

- Custom User Model 을 만들기 위해서는 BaseUserManager, AbstractUser 두 개의 클래스가 필요하다.

 

- BaseUserManager

    User 를 생성할 때 사용하는 헬퍼 클래스

    모델을 관리하는 클래스

    user 생성할 때의 행위를 지정

    모든 django model 들은 Manager 를 통해서 QuerySet 을 받는다. DB에서 query를 처리할 때,  Manager 를 무조건 거쳐야 한다. 

 

- AbstractUser

    실제 모델이 상속받아 생성하는 클래스

    반드시 objects 값을 통해 헬퍼 클래스를 지정해야한다.

 

 

 

 

user/models.py/AbstractUser

GENDER_CHOICES = (
    (0, 'Female'),
    (1, 'Male'),
    (2, 'Not to disclose')
)

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 = []

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

objects : 헬퍼 클래스 지정

USERNAME_FIELD : 유저 모델의 Unique Identifier. unique = True 가 옵션으로 설정된 필드 값으로 설정한다. 

REQUIRED_FIELDS : 필수로 받고 싶은 필드 값. USERNAME_FIELD 값과 패스워드는 항상 기본적으로 요구하기 때문에 따로 명시하지 않아도 된다.

 

 

 

 

user/models.py/BaseUserManager

class UserManager(BaseUserManager):
    def _create_user(self, email, username, password, gender = 1, **extra_fields):
        if not email :
            raise ValueError('The given email mist 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, username, password, **extra_fields)

 

create_user : 일반 유저 생성

일반 사용자를 생성한다. is_staff , is_superuser 값에 False를 취해서 스태프도 슈퍼유저도 아닌 그냥 일반 사용자임을 지정한다. 필드값을 지정하고 _create_user로 보낸다.

 

create_superuser : 관리자 유저 생성

is_staff, is_superuser 값에 True 값을 주어서, 관리자에게는 스태프, 슈퍼유저의 권한을 모두 갖도록 한다. 혹시 두 값이 False 라면, 에러 메세지를 띄운다. 관리자 역시 _create_user 로 보낸다.

 

_create_user : 유저 생성

파라미터로 전달받은 필드값들을 user 객체에 저장하고 DB에 저장한다. 

normalize : 중복 최소화를 위한 정규화 

 

 

user model을 AbstractUser를 상속할 때는 반드시 settings.py 에서 AUTH_USER_MODEL = '폴더명.클래스명' 값을 추가해야 migrate 가 제대로 진행된다.

AUTH_USER_MODEL = 'user.User'

python manage.py makemigrations user
python manage.py migrate user

 

 

 

 

python manage.py migrate 오류

 

(1) 오류 내용django.db.migrations.exceptions.InconsistentMigrationHistory: Migration admin.0001_initial is applied before its dependency user.0001_initial on database 'default'.

 

(2) 에러 해결

해당하는 app 폴더의 migrations 폴더와 db.sqlite3 파일을 삭제하고 다시 makemigrations 부터 하면 됨

 

 

user/admin.py

from django.contrib import admin
from .models import User

admin.site.register(User)