Django / Django Rest Framework - Как разрешить сериализатору модели установить в поле внешнего ключа модели значение NULL? - PullRequest
0 голосов
/ 06 июля 2019

Я хочу иметь возможность установить поле borrower (внешний ключ) в моей модели Asset на NULL, но, похоже, я не могу заставить его работать. Я пытаюсь отправить запрос PATCH с данными JSON, ключ borrower которого равен значению NULL, но поле borrower не будет обновлено до NULL для экземпляра модели. Возможно, существует проблема с сериализатором, из-за которого поле внешнего ключа не может быть установлено в NULL?

Я уже пытался передать allow_null=True в BorrowSerializer класс, но это не сработало. Я искал все выше и ниже в StackOverflow сообщения с похожими проблемами и решениями, но ничего из того, что я пробовал, не помогло.

Вот мои модели.py:

from django.conf import settings
from django.db import models
from django.utils import timezone
from datetime import date
from django.contrib.auth.models import User
from django.urls import reverse

import uuid

class Category(models.Model):
    """Model representing an Asset category"""

    name = models.CharField(max_length=128)

    def __str__(self):
        return self.name

class Borrower(models.Model):
    first_name = models.CharField(max_length=64)
    last_name = models.CharField(max_length=128)
    associated_user = models.ForeignKey(User, on_delete=models.CASCADE, null=True)

    def __str__(self):
        return f'{self.first_name} {self.last_name}'

class Asset(models.Model):
    """Model representing an Asset"""
    # Unique identifier for an instance of an asset (a barcode of sorts)
    uid = models.UUIDField(primary_key=True, default=uuid.uuid4)
    name = models.CharField(max_length=200)
    manufacturer = models.CharField(max_length=64)
    model = models.CharField(max_length=128)
    description = models.TextField()
    category = models.ManyToManyField(Category)
    owner = models.ForeignKey(User, on_delete=models.CASCADE)
    borrower = models.ForeignKey(Borrower, on_delete=models.CASCADE, null=True, blank=True)
    checked_out = models.BooleanField(default=False)
    return_date = models.DateField(null=True, blank=True)    

    CONDITION_TYPE = (
        ('e', 'Excellent'),
        ('g', 'Good'),
        ('f', 'Fair'),
        ('p', 'Poor'),
    )

    condition = models.CharField(
        max_length=1,
        choices=CONDITION_TYPE,
        blank=True,
        help_text='Asset condition')

    class Meta:
            ordering = ['return_date']

    @property
    def is_dueback(self):
        if self.return_date and date.today() > self.return_date:
            return True
        return False

    def display_category(self):
        """Create a string for the Category. This is required to display category in Admin."""
        return ', '.join(category.name for category in self.category.all())

    display_category.short_description = 'Category'

    def __str__(self):
        return f'{self.uid} - {self.name}'

    def get_absolute_url(self):
        return reverse('asset-detail', args=[str(self.uid)])

Вот мой файл serializers.py:

from rest_framework import serializers
from inventory.models import Asset, Borrower, Category

class BorrowerSerializer(serializers.ModelSerializer):
    class Meta:
        model = Borrower
        fields = ('first_name',
                  'last_name',
                  'associated_user'
        )

    def update(self, instance, validated_data):
        print('Update method triggered.')
        instance.first_name = validated_data.get('first_name', instance.first_name)
        instance.last_name = validated_data.get('last_name', instance.last_name)
        instance.associated_user = validated_data.get('associated_user'. instance.associated_user)
        instance.save()
        return instance

class CategorySerializer(serializers.ModelSerializer):
    class Meta:
        model = Category
        fields = '__all__'

class AssetSerializer(serializers.ModelSerializer):
    borrower = BorrowerSerializer(allow_null=True)
    category = serializers.StringRelatedField(many=True)
    condition = serializers.CharField(source='get_condition_display')

    class Meta:
        model = Asset
        fields = ('name', 
                  'manufacturer', 
                  'model',
                  'description',
                  'condition',
                  'category',
                  'borrower',
                  'checked_out',
                  'return_date',
                  'is_dueback',
        )

Вот мои DRF API Просмотров:

class AssetRetrieveUpdateDestroy(RetrieveUpdateDestroyAPIView):
    lookup_field = 'uid'
    serializer_class = AssetSerializer

    def get_queryset(self):
        user = self.request.user
        return Asset.objects.filter(owner=user)

class BorrowerRetrieveUpdateDestroy(RetrieveUpdateDestroyAPIView):
    lookup_field = 'id'
    serializer_class = BorrowerSerializer

    def get_queryset(self):
        return Borrower.objects.all()

Я ожидал, что поле borrower в моем экземпляре модели Asset будет обновлено до NULL, когда я передам этот JSON:

data = {
   'borrower': null
}

Однако мой экземпляр модели не будет обновлять поле borrower, когда это внешний ключ. Работает нормально, если поле CharField или что-то еще, но на внешних ключах. Я распечатал данные, полученные из моего запроса AJAX, но поле borrower осталось без изменений.

Заранее благодарим за любую помощь, которую вы можете оказать.

1 Ответ

1 голос
/ 07 июля 2019

Из документов :

Если вы поддерживаете вложенные представления, доступные для записи, вам нужно написать .create() или .update() методы, которые обрабатывают сохранение нескольких объектов.

Таким образом, вам необходимо реализовать метод .update() в вашем AssetSerializer:

class AssetSerializer(serializers.ModelSerializer):
    borrower = BorrowerSerializer(allow_null=True)
    ...

    class Meta:
        model = Asset
        fields = (...)

    def update(self, instance, validated_data):
        instance.borrower = validated_data.get('borrower')
        instance.save()

        return instance

Это должно сделать эту работу.Поэтому, независимо от того, что вы передали allow_null=True, он примет нулевые значения , но не обновит ваши вложенные отношения, если вы не определите свой собственный механизм обновления.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...