Search code examples
pythondjangodjango-rest-frameworkdjango-rest-viewsets

How to create / update 2 objects [Django Rest Framework]


I'm new to django rest framework and I'm trying to create / update 2 objects in the same request:

class PessoaSerializer(serializers.ModelSerializer):
    user = UserSerializer(required=False)

    class Meta:
        model = Pessoa
        fields = ('id', 'user', 'nome', 'email', 'cargo', 'fone', 'cep', 'celular', 'endereco', 'observacao', 'municipio')

    def create(self, validated_data):

        usuario = User()
        usuario.first_name = validated_data.get('nome')
        usuario.username = validated_data.get('email')
        usuario.last_name = validated_data.get('nome') #TODO
        usuario.email = validated_data.get('email')
        usuario.is_active = True
        usuario.is_staff = False
        usuario.is_superuser = False
        usuario.set_password(validated_data.get('email')) #TODO
        usuario.last_login = datetime.datetime.now()
        usuario.date_joined = datetime.datetime.now()
        usuario.save()

        validated_data['user'] = usuario

        return Pessoa.objects.create(**validated_data)

    def update(self, instance, validated_data):
       #TODO

This is the best way to to it? Or I'm missing something?


Solution

  • try below code:

    def create(self, validated_data):
        user_data = validated_data.pop('user')
        pessoa = Pessoa.objects.create(**validated_data)
        pessoa.user.add(user_data)
        return pessoa