Python 如何向rest身份验证的注册端点添加额外字段
我正在使用RESTAUTH注册api进行用户注册。我在Python 如何向rest身份验证的注册端点添加额外字段,python,django,django-rest-framework,Python,Django,Django Rest Framework,我正在使用RESTAUTH注册api进行用户注册。我在UserProfile模型中有一些额外的字段 from django.db import models from django.contrib.auth.models import User from django.db.models.signals import post_save class UserProfile(models.Model): user = models.OneToOneField(User, on_dele
UserProfile
模型中有一些额外的字段
from django.db import models
from django.contrib.auth.models import User
from django.db.models.signals import post_save
class UserProfile(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE)
org_id = models.CharField(max_length=100, default='')
is_teacher = models.BooleanField(blank=True, default=False)
def __str__(self):
return self.user.username
def create_profile(sender, **kwargs):
if kwargs['created']:
user_profile = UserProfile.objects.create(user=kwargs['instance'])
post_save.connect(create_profile, sender=User)
UserProfile
模型如上图所示。如何将这些字段添加到rest auth regestration api端点并将数据保存到数据库。我自己找到了答案
序列化程序可以编写为
from rest_framework import serializers
from rest_auth.registration.serializers import RegisterSerializer
from .models import UserProfile
class RegistrationSerializer(RegisterSerializer):
first_name = serializers.CharField(required=False)
last_name = serializers.CharField(required=False)
personal_id = serializers.CharField(required=True)
def custom_signup(self, request, user):
user.first_name = self.validated_data.get('first_name', '')
user.last_name = self.validated_data.get('last_name', '')
user.userprofile.personal_id = self.validated_data.get(
'personal_id', '')
user.save(update_fields=['first_name', 'last_name'])
user.userprofile.save(update_fields=['org_id'])
我没有添加is\u teacher
,因为它是可选的
在views.py中,扩展rest\u auth.regeistration.views的RegisterView
,以传递此数据及其完成
class RegistrationView(RegisterView):
serializer_class = RegistrationSerializer
最后添加一个url并传递RegisterView.as\u view()