[Django - Airbnb 클론 코딩 #3] USER 데이터 모델링

USER 데이터 모델링하기

users 애플리케이션의 models.py에 Airbnb 사용자들의 데이터 모델을 담당할 User 클래스를 생성해 준다.

Django에서는 자체적으로 USER 데이터 모델을 제공해 준다. 이 데이터 모델을 그대로 사용해도 되지만 Airbnb의 USER 모델이 되기엔 부족하기 때문에 커스텀을 해주어야 한다. 

 

USER MODEL 커스텀하기

 

1. setttings.py에 AUTH_USER_MODEL = "myapp.MyUser" 등록
2. models.py에서 Django에서 제공하는 User 모델(AbstractUser)을 상속

USER Model 커스텀은 공식문서를 따라서 진행하였다.

AUTH_USER_MODEL에는 프로젝트의 USER 데이터 모델을 담당하는 클래스를 지정해주면 된다. 해당 프로젝트의 USER 데이터 모델을 담당하는 클래스는 users 애플리케이션의 User 클래스이므로 "users.User"로 등록해주었다.

해당 프로젝트의 USER 데이터 모델은 Django에서 제공하는 USER 데이터 모델에 프로젝트에 알맞은 데이터 필드를 추가해주려는 것이기 때문에 AbstratcUser 클래스를 상속받아서 USER 클래스를 생성해주었다.

 

 

USER MODEL 데이터 필드 추가하기

 

Django의 데이터 모델의 데이터 필드의 종류는 공식문서에서 확인할 수 있다.

아래와 같이 데이터 필드들을 추가해주었다. 

User 모델의 추가된 데이터 필드

1) 데이터 필드가 정의되려면 레코드의 초기 상태가 정의되어야 한다.

 

FieldType 선언만 해줘도 빈 레코드로써 초기 상태가 정의될 수 있다. 하지만 DateField 같은 몇몇 FieldType들은 따로 초기 상태를 정의해줘야 하므로 옵션을 사용해줘야 한다. 초기 상태를 정의하는 데에는 두 가지 옵션 이 존재한다. 

 

1. default 인자 : default 값을 설정해 준다.
2. null = True : 레코드가 null 값임을 허용해 준다.

2) blank = True 옵션은 Admin 패널에서 해당 필드가 입력되지 않아도 객체를 생성할 수 있게 해 준다.

 

개발 중일 때는 해당 모델의 데이터들을 Admin 패널에서 직관적으로 입력해줄 수 있다. 이때, 레코드가 필수적으로 입력되어줘야 할 데이터 필드와 그러지 않아도 되는 데이터 필드가 있을 수 있다. 하지만 따로 설정을 안 해주면 모든 필드들이 입력되지 않았을 시 입력되지 않았다는 경고 문구가 뜨게 된다. 

 

3) choices 인자는 Admin 패널에서 Select Box 역할을 한다.

 

models.CharField()는 html의 <select>와 같은 choices 인자가 존재한다. 이때, 실제 데이터베이스에 들어갈 데이터와 admin 패널에 보일 데이터 둘 다 명시되어야 한다. 아래 예시와 같이 choices = LANGUAGE_CHOICES을 통해 사용자로 하여금 사용하는 언어를 Select Box로 선택할 수 있게 하였다.

 

LANGUAGE_ENGLISH = "en"
LANGUAGE_KOREAN = "kr"
LANGUAGE_CHOICES = ((LANGUAGE_ENGLISH, "English"), (LANGUAGE_KOREAN, "Korean"))
language = models.CharField(choices=LANGUAGE_CHOICES, max_length=2, blank=True)

공식문서 참조

LANGUAGE_CHOICES 변수에서 내부 튜플의 첫 번째 원소는 DB에 들어갈 데이터이고, 두 번째 원소는 Admin 패널에 보일 데이터이다.

 

 

Admin 패널에서의 USER MODEL

User 클래스에 커스텀 된 USER 데이터 모델을 admin.py에 등록해준다. 등록 방법은 공식문서에 2가지 방법이 나오는데 이 중 Register Decorator을 사용했다. 또한, 공식 문서대로 models.ModelAdmin을 상속받아서 사용하지 않고 USER 모델용 Admin 패널인 UserAdmin을 상속받아서 사용했다. 

 

from django.contrib.auth.admin import UserAdmin
from . import models

@admin.register(models.User)
class CustomUserAdmin(UserAdmin):
    pass

USER 모델의 Admin 패널 커스텀은 나중에 다루도록 하겠다.