我在 Django rest 框架上使用 django_rest_passwordreset 进行身份验证。
我按照他们官方文档上的教程进行了登录,注册效果很好。
登录返回访问和刷新令牌。我想要做的也是返回用户对象以及访问和刷新令牌。
我有一个自定义模型
# auth/models.py
# Inherited from user
class MUser(AbstractUser, models.Model):
email = models.EmailField(unique=True)
role = models.CharField(max_length=40, default="student")
USERNAME_FIELD = 'email'
EMAIL_FIELD = 'email'
REQUIRED_FIELDS = ['username', 'role']
序列化器
# auth/serializers.py
class TokenObtainPairSerializer(TokenObtainSerializer):
@classmethod
def get_token(cls, user):
return RefreshToken.for_user(user)
def validate(self, attrs):
data = super().validate(attrs)
refresh = self.get_token(self.user)
data['refresh'] = str(refresh)
data['access'] = str(refresh.access_token)
if api_settings.UPDATE_LAST_LOGIN:
update_last_login(None, self.user)
return data
我的观点
# auth/views.py
class TokenObtainPairView(TokenViewBase):
"""
Takes a set of user credentials and returns access and refresh JSON web
token pair to prove the authentication of those credentials.
"""
serializer_class = serializers.TokenObtainPairSerializer
总之,登录成功后,我收到
{
access: xxx,
refresh: xxx,
# I want to add this
user: {username, first_name, last_name, email, ...}
}
答案 0 :(得分:0)
我想这可能对您有所帮助:
def validate(self, attrs):
data = super().validate(attrs)
refresh = self.get_token(self.user)
data['refresh'] = str(refresh)
data['access'] = str(refresh.access_token)
@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@
my_user = User.objects.filter(pk=self.user.id).first()
if my_user:
# use user serelizor or parse required fields
data['user'] = my_user
@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@@
if api_settings.UPDATE_LAST_LOGIN:
update_last_login(None, self.user)
return data