想使用输入电子邮件设置自定义用户的用户名,但是在哪里进行自定义分配?同时它也接收一个文件。
Models.py
class CustomUser(AbstractUser):
avatar = models.ImageField(max_length=None, upload_to='avatar', blank=True)
Serializers.py
class CustomUserSerializer(serializers.ModelSerializer):
class Meta:
model = CustomUser
fields = ('id', 'first_name', 'last_name', 'email', 'password', 'avatar', 'groups')
Views.py
class CustomUserViewSet(viewsets.ModelViewSet):
queryset = CustomUser.objects.all()
serializer_class = CustomUserSerializer
提前谢谢你。
@Anzel所说的会起作用,但如果你想在 django-rest-framework 中做到这一点,你可以覆盖你的CustomUserSerializer
的create
方法。喜欢:
class CustomUserSerializer(serializers.ModelSerializer):
groups = serializers.PrimaryKeyRelatedField(many=True, read_only=True)
def create(self, validated_data):
user = CustomUser.objects.create_user(
username =validated_data['email'], # HERE
email =validated_data['email'],
password =validated_data['password'],
first_name =validated_data['first_name'],
last_name =validated_data['last_name'],
avatar =validated_data['avatar'],
)
user.groups = validated_data['groups']
return user
class Meta:
model = CustomUser
fields = (
'id',
'first_name',
'last_name',
'email',
'password',
'avatar',
'groups'
)