Django 커스텀 User 모델의 완전한 구현과 주의사항

자, Django에서 커스텀 User 모델 만드는 거, 막막하게 느껴지시죠? 저도 처음엔 엄청 힘들었어요. 기본 User 모델만으론 부족하고, 뭘 어떻게 해야 할지 몰라서 며칠 밤을 새운 기억도 나네요. 하지만 이제는 좀 익숙해져서 여러분께 쉽게 설명해 드릴 수 있을 것 같아요!

Django 기본 User 모델은, 음… 정말 기본적인 것만 제공하거든요. 실제 서비스를 만들려면 추가 정보, 예를 들어 전화번호나 주소 같은 게 필요하잖아요? 그럴 때 커스텀 User 모델을 만들면 딱 맞는 사용자 관리 시스템을 만들 수 있어요. 게다가 나중에 기능 추가하거나 수정하기도 훨씬 수월해지죠. 안 그러면 나중에 후회할지도 몰라요… 제가 그랬거든요. 😅

자, 그럼 핵심 개념부터 짚어볼까요? AbstractBaseUser랑 AbstractUser라는 녀석들이 있는데, 이 둘 중 하나를 상속받아서 커스텀 모델을 만들어야 해요. AbstractUser는 기본적인 정보(username, password, email 등)를 다 가지고 있어서 편리하긴 한데, 필요 없는 정보가 있을 수도 있어요. AbstractBaseUser는 좀 더 유연하게 원하는 정보만 골라서 쓸 수 있지만, 처음엔 조금 복잡하게 느껴질 수도 있고요. 어떤 걸 써야 할지는 여러분 프로젝트의 필요에 따라 달라요. 저는 보통 프로젝트 규모가 작으면 AbstractUser를, 좀 더 유연성이 필요하면 AbstractBaseUser를 선택하는 편이에요.

그리고 중요한 게 하나 더 있는데, username 대신 email을 사용자 식별자로 쓰고 싶다면 USERNAME_FIELD 설정을 바꿔줘야 해요. 이 부분 꼭 기억해두세요! 그리고 PermissionsMixin도 같이 사용하는 걸 추천해요. 사용자 권한 관리를 훨씬 쉽게 해주거든요.

자, 이제 실제 코드를 보여드릴게요. 아래 코드는 제가 실제로 사용했던 코드인데, 여러분은 이걸 바탕으로 필요에 맞게 수정하면 돼요.

from django.db import models
from django.contrib.auth.models import AbstractUser, PermissionsMixin
from django.utils.translation import gettext_lazy as _

class CustomUser(AbstractUser, PermissionsMixin):
    """Custom user model with additional fields."""
    email = models.EmailField(_('email address'), unique=True)
    phone_number = models.CharField(max_length=20, blank=True)
    address = models.CharField(max_length=255, blank=True)
    is_staff = models.BooleanField(default=False)
    is_active = models.BooleanField(default=True)

    USERNAME_FIELD = 'email'
    REQUIRED_FIELDS = ['username']  # username 필드는 선택적이지만, 유지하는 것이 좋습니다.

    def __str__(self):
        return self.email

이 코드를 어떤 앱에 넣을지 정하셨으면, settings.py 파일에서 AUTH_USER_MODEL 설정을 바꿔줘야 해요. 여기서 실수하면 안 되니까 꼼꼼하게 확인하세요! 예를 들어, 제가 만든 앱 이름이 myapp이라면 다음과 같이 적으면 됩니다.

AUTH_USER_MODEL = 'myapp.CustomUser'

그리고 마지막으로, python manage.py makemigrations 와 python manage.py migrate 명령어를 실행해서 데이터베이스에 반영해야 해요. 이 부분은 절대 빼먹으면 안 돼요! 데이터베이스 마이그레이션은 중요해요. 저는 한번 실수해서 데이터 날릴 뻔 했거든요… 😨

몇 가지 주의사항을 더 말씀드리자면, AbstractUser와 AbstractBaseUser 선택, USERNAME_FIELD 설정, PermissionsMixin 사용, 그리고 데이터베이스 마이그레이션 실행과 테스트를 꼼꼼하게 하는 게 정말 중요해요. 특히 테스트는 꼭 하세요! 나중에 문제 발생하면 디버깅하기 어려워요.

결론적으로, Django 커스텀 User 모델은 프로젝트의 핵심이라고 할 수 있어요. 이 가이드가 여러분의 Django 프로젝트에 도움이 되었으면 좋겠고, 궁금한 점이나 어려운 부분이 있으면 언제든지 질문해주세요! 함께 고민해봐요! 😊