# users/models.py
from django.db import models
from django.contrib.auth.models import BaseUserManager, AbstractBaseUser
# 나이 최솟값, 최댓값을 설정하기 위해 import함
from django.core.validators import MinValueValidator, MaxValueValidator
class UserManager(BaseUserManager):
def create_user(self, email, name, gender, age, introduction, password=None):
if not email:
raise ValueError("Users must have an email address")
user = self.model(
email=self.normalize_email(email),
name=name,
gender=gender,
age=age,
introduction=introduction,
)
user.set_password(password)
user.save(using=self._db)
return user
def create_superuser(self, email, password=None):
user = self.create_user(
email=email,
password=password,
name="관리자",
gender="non-choice",
age="0",
introduction="관리자 계정입니다"
)
user.is_admin = True
user.save(using=self._db)
return user
class User(AbstractBaseUser):
GENDER_CHOICES = (
("male", "남자"),
("female", "여자"),
("unknown", "모름"),
("non-choice", "선택하지 않음")
)
email = models.EmailField(
verbose_name="email address",
max_length=255,
unique=True,
)
name = models.CharField("이름", max_length=20)
gender = models.CharField("성별", max_length=20, choices=GENDER_CHOICES)
age = models.IntegerField(
"나이", validators=[MinValueValidator(0), MaxValueValidator(120)])
introduction = models.TextField("자기소개", blank=True)
is_active = models.BooleanField(default=True)
is_admin = models.BooleanField(default=False)
objects = UserManager()
USERNAME_FIELD = "email"
# 필드에 blank=True, null=True를 하지 않으면 필수값이 되던데 이 부분이 왜 따로 있는지 궁금
REQUIRED_FIELDS = []
def __str__(self):
return self.email
def has_perm(self, perm, obj=None):
"Does the user have a specific permission?"
# Simplest possible answer: Yes, always
return True
def has_module_perms(self, app_label):
"Does the user have permissions to view the app `app_label`?"
# Simplest possible answer: Yes, always
return True
@property
def is_staff(self):
"Is the user a member of staff?"
# Simplest possible answer: All admins are staff
return self.is_admin
- urls.py
# users/urls.py
from django.urls import path
from users import views
urlpatterns = [
path('api/signup/', views.SignupView.as_view(), name='signup_view'),
]
- serializers.py
# users/serializers.py
from rest_framework import serializers
from users.models import User
class UserSerializer(serializers.ModelSerializer):
class Meta:
model = User
fields = "__all__"
def create(self, validated_data):
user = super().create(validated_data)
password = user.password
user.set_password(password)
user.save()
return user
- views.py
# users/views.py
from rest_framework import status
from rest_framework.views import APIView
from rest_framework.response import Response
from users.serializers import UserSerializer
class SignupView(APIView):
def post(self, request):
"""회원가입"""
serializer = UserSerializer(data=request.data)
if serializer.is_valid():
serializer.save()
return Response({"message": "회원가입이 완료되었습니다!"}, status=status.HTTP_201_CREATED)
else:
return Response({"message": f"{serializer.errors}"}, status=status.HTTP_400_BAD_REQUEST)
<로그인>
로그인은 어렵지 않았다
simpleJWT 방식을 이용했는데 강의 때 배운 걸 써먹고 싶었다
- urls.py
# users/urls.py
from django.urls import path
from users import views
from rest_framework_simplejwt.views import (
TokenRefreshView,
)
urlpatterns = [
...
path('api/token/', views.CustomTokenObtainPairView.as_view(),
name='token_obtain_pair'),
path('api/token/refresh/', TokenRefreshView.as_view(), name='token_refresh'),
]
