This commit is contained in:
2025-04-22 18:40:18 +09:00
parent 386ec64ae0
commit d8c35e5d4f
5 changed files with 60 additions and 9 deletions

View File

@ -1,6 +1,6 @@
from rest_framework import serializers
from .models import CustomUser
from rest_framework_simplejwt.serializers import TokenObtainPairSerializer
class RegisterSerializer(serializers.ModelSerializer):
password = serializers.CharField(write_only=True)
@ -15,3 +15,20 @@ class RegisterSerializer(serializers.ModelSerializer):
user.set_password(password)
user.save()
return user
class CustomTokenObtainPairSerializer(TokenObtainPairSerializer):
@classmethod
def get_token(cls, user):
token = super().get_token(user)
# ✅ JWT payload에 커스텀 정보 추가
token["name"] = user.name
token["grade"] = user.grade
token["email"] = user.email # 선택적으로 추가 가능
return token
# def validate(self, attrs):
# data = super().validate(attrs)
# data['name'] = self.user.name # ✅ 여기에 토큰 응답 추가 정보 삽입
# return data