Вставка обновления DRF в моделях отношений ManytoMany, ForeignKey

У меня есть шесть моделей, они приведены ниже:

class Certificate(DateTimeLog):
    name = models.TextField(max_length=255)


class Vacancy(DateTimeLog):
    name = models.CharField(max_length=255)
    parent_position = models.ForeignKey("self", on_delete=models.CASCADE, null=True, blank=True,
                                        related_name='sub_positions')


class Region(DateTimeLog):
    name = models.CharField(max_length=255)


class MaritalStatus(DateTimeLog):
    name = models.CharField(max_length=255)


class Candidate(DateTimeLog):
    pin = models.CharField(max_length=16, unique=True)
    first_name = models.CharField(max_length=64, blank=True, null=True)
    last_name = models.CharField(max_length=64, blank=True, null=True)
    marital_status = models.ForeignKey(MaritalStatus, on_delete=models.SET_NULL, null=True, blank=True)
    certificate = models.ManyToManyField(Certificate, blank=True)


class Candidacy(DateTimeLog):
    candidate = models.ForeignKey(Candidate, on_delete=models.CASCADE)
    vacancy = models.ForeignKey(Vacancy, on_delete=models.CASCADE)
    work_region = models.ForeignKey(Region, on_delete=models.SET_NULL, null=True, blank=True)

Теперь я хочу обработать, если запись кандидата существует (я проверяю ее с помощью пин), то проверить и обновить данные, связанные с кандидатом. Если кандидат не существует, создайте его. После создания или обновления кандидата присваиваем его кандидатуре. Мой сериализатор выглядит следующим образом:

class CandidateSerializer(serializers.ModelSerializer):
    marital_status = MaritalStatusSerializer(required=False)
    certificate = CertificateSerializer(many=True, required=False)

    def create(self, validated_data):
        marital_status_data = validated_data.pop("marital_status")
        certificate_data = validated_data.pop("certificate")
        candidate = Candidate.objects.create(**validated_data)

        ms_o = MaritalStatus.objects.get(name=marital_status_data["name"])
        candidate.marital_status = ms_o

        for certificate in certificate_data:
            certificate_o = Certificate.objects.create(**certificate)
            candidate.certificate.add(certificate_o)

        candidate.save()
        return candidate

    class Meta:
        model = Candidate
        fields = '__all__'
        depth = 1


class CandidacySerializer(serializers.ModelSerializer):
    candidate = CandidateSerializer()
    vacancy = VacancySerializer()
    work_region = RegionSerializer()

    def create(self, validated_data):
        candidate_s = CandidateSerializer()
        candidate_data = validated_data.pop('candidate')

        vacancy_data = validated_data.pop('vacancy')
        work_region_data = validated_data.pop('work_region')

        vac_o = Vacancy.objects.get(name=vacancy_data['name'])
        wr_o = Region.objects.get(name=work_region_data['name'])
        candidate_o = candidate_s.create(validated_data=candidate_data)
        validated_data.update({
            'candidate': candidate_o,
            'vacancy': vac_o,
            'work_region': wr_o
        })
        candidacy = Candidacy.objects.create(**validated_data)
        return candidacy

    class Meta:
        model = Candidacy
        fields = '__all__'
        depth = 1

В настоящее время я могу создать его с помощью post-запроса. Нужно ли мне проверять его явно или это можно сделать внутри сериализатора?

поэтому у вас есть более простой вариант. Существует функция get или create, которая проверит, существует ли объект, и вернет его, в противном случае создаст его. Подобный вопрос можно найти здесь. Другой вариант использования того же вопроса. Кстати, если этот пин является чувствительным, я рекомендую зашифровать его.

Поскольку вы уже сделали пин уникальным с

pin = models.CharField(max_length=16, unique=True)

нет никаких шансов иметь два одинаковых пина. и теперь, чтобы проверить кандидата на основе пина, вам придется вручную сделать что-то вроде этого в вашем методе create.

if Candidate.objects.filter(pin = revievedPinData):
    here candidate is already present so update the information information
    and again check if Canditate is in Candidacy or not
    first get the candidate object
    c = Candidate.objects.get(pin = revievedPinData)
    then
    if Candidacy.objects.filter(candidate = c):
       if true candidate is already assigned and the updated value of candidate will reflect here
    else: create the Candidacy with the current candidate and other values
else:
    if candidate is not there (from the given pin)
    create a new candidate
    aa = candidate.objects.create()
    create a new Candidacy with candidate aa
    Candidacy.objects.create(candidate=aa, **otherdata)

Прежде всего, вы можете использовать update_or_create в своих CandidateSerializer следующим образом:

class CandidateSerializer(serializers.ModelSerializer):
    marital_status = MaritalStatusSerializer(required=False)
    certificate = CertificateSerializer(many=True, required=False)

    def create(self, validated_data):
        marital_status_data = validated_data.pop("marital_status")
        certificate_data = validated_data.pop("certificate")
        candidate, created = Candidate.objects.update_or_create(
            pin=validated_data['pin'],
            **validated_data
        )

        ms_o = MaritalStatus.objects.get(name=marital_status_data["name"])
        candidate.marital_status = ms_o

        for certificate in certificate_data:
            certificate_o = Certificate.objects.create(**certificate)
            candidate.certificate.add(certificate_o)

        candidate.save()
        return candidate

    class Meta:
        model = Candidate
        fields = '__all__'
        depth = 1

Во-вторых, вы не должны вызывать метод create() напрямую. Вместо него следует использовать save(). Вы должны заменить:

candidate_s = CandidateSerializer()
candidate_o = candidate_s.create(validated_data=candidate_data)

by

candidate_s = CandidateSerializer(data=candidate_data)
candidate_o = candidate_s.save()

Вы можете прочитать больше о сохранении экземпляров здесь.

Вернуться на верх