[Django] User 모델 확장

안녕하세요🙏 개발자 재민콩입니다.

 

오늘은 Django에서 제공하는 User 모델을 확장하는 방법에 대해 알아보겠습니다.

Model에 AbstractUser를 사용하여 간단하게 모델을 확장할 수 있습니다.

 

1. models.py 추가

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

class User(AbstractUser):
    nickname = models.CharField(_("닉네임"), max_length=100, blank=True, null=True)

 

2. settings.py 추가

INSTALLED_APPS = [
    ...
    'project',
]

AUTH_USER_MODEL = 'project.User'

 

3. serializers.py 추가

from rest_framework import serializers
from .models import User

class UserInfoSerializer(serializers.ModelSerializer) :
    class Meta:
        model = User
        fields = (
            'username',	// 기존 User에서 제공
            'email',	// 기존 User에서 제공
            'nickname', // 새로 추가한 컬럼
        )

 

3가지 python 파일만 수정하게 되면 간단하게 User 모델을 확장할 수 있습니다!