서버(Server)/장고 (Django)

Django AbstractUser 불필요한 필드 지우는 방법

2023. 1. 4. 20:57
목차
  1. Django의 유저 모델들
  2. AbstractUser 모델
728x90

Django의 유저 모델들

Django에서는 기본적으로 유저 모델 django.contrib.auth.models의 User를 제공하고, 이를 통해 간편하게 인증 등을 처리할 수 있다. 또한 기본제공되는 유저모델을 확장할 수 있게 다음과 같은 다양한 방법이 있다.

  • User에 기반한 Proxy Model 생성하기
  • 1:1 로 User에 연결된 모델 만들기
  • AbstractUser 상속받아 커스텀 유저 모델 만들기
  • AbstractBaseUser 상속받아 커스텀 유저 모델 만들기

AbstractUser 모델

기본적으로 제공하는 django.contrib.auth.models의 User을 사용하더라도, 이후에 유저 모델을 수정할 일이 생길수있다. 그런데 이후에 이를 수정한다면 그 과정이 매우 힘들기에, 기본적으로 AbstractUser를 상속한 유저를 다음과 같이 새로 만들어 사용하는것을 추천한다. (기본제공하는 User도 그냥 AbstractUser를 상속받기만 한 형태이다.)

# models.py
from django.contrib.auth.models import AbstractUser

class User(AbstractUser):
    pass

다음과 같이 선언만 해두더라도, 이후에 수정하기 용이하기에 프로젝트 시작 시 유용하다.

AbstractUser는 다음과 같이 정의되어 있다.

# django.contrib.auth.models
class AbstractUser(AbstractBaseUser, PermissionsMixin):
    """
    An abstract base class implementing a fully featured User model with
    admin-compliant permissions.

    Username and password are required. Other fields are optional.
    """

    username_validator = UnicodeUsernameValidator()

    username = models.CharField(
        _("username"),
        max_length=150,
        unique=True,
        help_text=_(
            "Required. 150 characters or fewer. Letters, digits and @/./+/-/_ only."
        ),
        validators=[username_validator],
        error_messages={
            "unique": _("A user with that username already exists."),
        },
    )
    first_name = models.CharField(_("first name"), max_length=150, blank=True)
    last_name = models.CharField(_("last name"), max_length=150, blank=True)
    email = models.EmailField(_("email address"), blank=True)
    is_staff = models.BooleanField(
        _("staff status"),
        default=False,
        help_text=_("Designates whether the user can log into this admin site."),
    )
    is_active = models.BooleanField(
        _("active"),
        default=True,
        help_text=_(
            "Designates whether this user should be treated as active. "
            "Unselect this instead of deleting accounts."
        ),
    )
    date_joined = models.DateTimeField(_("date joined"), default=timezone.now)

    objects = UserManager()

    EMAIL_FIELD = "email"
    USERNAME_FIELD = "username"
    REQUIRED_FIELDS = ["email"]

    class Meta:
        verbose_name = _("user")
        verbose_name_plural = _("users")
        abstract = True

    def clean(self):
        super().clean()
        self.email = self.__class__.objects.normalize_email(self.email)

    def get_full_name(self):
        """
        Return the first_name plus the last_name, with a space in between.
        """
        full_name = "%s %s" % (self.first_name, self.last_name)
        return full_name.strip()

    def get_short_name(self):
        """Return the short name for the user."""
        return self.first_name

    def email_user(self, subject, message, from_email=None, **kwargs):
        """Send an email to this user."""
        send_mail(subject, message, from_email, [self.email], **kwargs)

username, email, first_name, last_name 등 다양한 필드가 기본적으로 정의되어 있는데, 이중에 사용하지 않을 필드도 여럿 있을 것이다. 이때 이런 필드들을 지우기 위해서는 다음과 같이 오버라이딩 해주면 된다.

# models.py
class User(AbstractUser):
    first_name = None
    last_name = None
    
    email = None
    EMAIL_FIELD = None
    REQUIRED_FIELDS = []

# 이메일 필드 삭제시에는 UserManager도 바꿔주어야 한다
# https://docs.djangoproject.com/en/4.1/topics/auth/customizing/#a-full-example

이렇게 대부분의 기능을 사용하지 않는다면 AbstractBaseUser에도 인증기능이 있기에 이를 상속받는 것도 추천한다.

728x90

'서버(Server) > 장고 (Django)' 카테고리의 다른 글

[Django Basic] 1.2. 장고 앱 배포하기  (0) 2023.07.15
[Django Basic] 1.1. 파이썬에서의 가상 환경  (0) 2023.07.15
[Django Basic] 1. Hello Django, Project, App  (0) 2023.07.06
[Django Basic] 0. Django Basic을 시작하며  (0) 2023.07.05
Django SECRET_KEY 새로 생성하기  (0) 2023.06.23
  • Django의 유저 모델들
  • AbstractUser 모델
Wibaek
생쥐 개발자
Wibaek
총 방문
오늘
어제
  • 전체보기 (111)
    • 서버(Server) (4)
      • 장고 (Django) (20)
      • 스프링 (Spring) (0)
    • 프론트엔드 (Frontend) (4)
    • 파이썬 (Python) (8)
    • 자바 (Java) (1)
    • 인프라 (Infra) (10)
    • 알고리즘 (Algorithm), PS (4)
      • Leetcode (0)
      • Baekjoon Online Judge (3)
    • CS (20)
      • 자료구조 (Data Structure) (19)
    • Troubleshooting (10)
    • 회고 & 기록 (10)
    • 기타 (Other) (12)
    • TIL (1)
    • 하나도 안 중요함 (6)

인기 글

250x250
hELLO · Designed By 정상우.
Wibaek
Django AbstractUser 불필요한 필드 지우는 방법
상단으로

티스토리툴바

단축키

내 블로그

내 블로그 - 관리자 홈 전환
Q
Q
새 글 쓰기
W
W

블로그 게시글

글 수정 (권한 있는 경우)
E
E
댓글 영역으로 이동
C
C

모든 영역

이 페이지의 URL 복사
S
S
맨 위로 이동
T
T
티스토리 홈 이동
H
H
단축키 안내
Shift + /
⇧ + /

* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.