Как получить поле «многие ко многим» с поиском обратных связей в сериализаторе REST Framework Django? - PullRequest
Новая
       4

Как получить поле «многие ко многим» с поиском обратных связей в сериализаторе REST Framework Django?

0 голосов
/ 30 января 2020

Пожалуйста, исправьте мой заголовок, если он не правильный. Моя проблема в том, что я хочу получить FinishType name из Product. Я пробовал 2 способа добиться этого: первая попытка и вторая попытка .
Мои упрощенные родственные модели в models.py: 

class Product(models.Model):
    product_id = models.CharField(max_length=6)
    color = models.ForeignKey(ColorParent, on_delete=models.SET_NULL, null=True)
    collection = models.ForeignKey(ProductCollection, on_delete=models.SET_NULL, null=True)

    @property
    def get_distributors(self):
        return Distributor.objects.filter(distributor__products=self).count()

    def __str__(self):
        return self.product_id

class FinishType(models.Model):
    name = models.CharField(max_length=50)

    def __str__(self):
        return self.name

class ProductFinishType(models.Model):
    product = models.ForeignKey(Product, on_delete=models.CASCADE)
    market = models.ForeignKey(Market, on_delete=models.CASCADE)
    finish_types = models.ManyToManyField(FinishType)

    def __str__(self):
        return '%s - %s' % (self.product, self.market)

class ProductAlias(models.Model):
    product = models.ForeignKey(Product, on_delete=models.CASCADE)
    market = models.ForeignKey(Market, on_delete=models.CASCADE)
    name = models.CharField(max_length=50, null=True, blank=True)

    def __str__(self):
        return '%s - %s' % (self.product, self.name)

My serializers.py: 

class ProductGridSerializer(serializers.ModelSerializer):
    name = serializers.SerializerMethodField(source='get_name')
    finishing = serializers.SerializerMethodField('get_finish_types')
    distributor = serializers.ReadOnlyField(source='get_distributors')

    @staticmethod
    def get_name(obj):
        return [pa.name for pa in obj.productalias_set.all()]

    @staticmethod
    def get_finish_types(obj):
        return [pft.name for pft in obj.productfinishtype_set.all().select_related('finish_types')]  # first attempt

    class Meta:
        model = Product
        fields = ['id', 'product_id', 'name', 'collection', 'finishing', 'distributor']

Первая попытка работает для поля name, которое извлекает ProductAlias s name, но дает мне эту ошибку:

FieldError at / api / product_grids /
Недопустимые имена полей, заданные в select_related: 'finish_types'. Выбор: продукт, рынок

My get_finish_types() со второй попытки: 

@staticmethod
def get_finish_types(obj):
    product_finish_types = obj.productfinishtype_set.all()
    response = ProductFinishTypeSerializer(product_finish_types, many=True, source='finish_types').data
    return response

Это дает мне все данные объекта: 

{
      "id": 1,
      "product_id": "BQ1111",
      "name": [
            "White Stone"
      ],
      "collection": 1,
      "finishing": [
          {
              "id": 1,
              "product": 1,
              "market": 1,
              "finish_types": [
                  1,
                  3,
                  5
              ]
          }
      ],
      "distributor": 5
},

My желаемый вывод что-то вроде: 

{
      "id": 1,
      "product_id": "BQ1111",
      "name": [
            "White Stone"
      ],
      "collection": 1,
      "finishing": [
            "Polished",
            "Carved",
            "Melted"
      ],
      "distributor": 5
},

Ответы [ 2 ]

1 голос
/ 30 января 2020

Создайте сериализатор для FinishType, 

class FinishTypeSerializer(serializers.ModelSerializer):
    class Meta:
        model = FinishType
        fields = ('name',)

и подключите его в ProductGridSerializer, используя SerializerMethodField

class ProductGridSerializer(serializers.ModelSerializer):
    name = serializers.SerializerMethodField(source='get_name')    
    distributor = serializers.ReadOnlyField(source='get_distributors')
    <b>finishing = serializers.SerializerMethodField()

    def get_finishing(self, product):
        qs = FinishType.objects.filter(productfinishtype__product=product)
        return FinishTypeSerializer(qs, many=True).data</b>

    @staticmethod
    def get_name(obj):
        return [pa.name for pa in obj.productalias_set.all()]

    class Meta:
        model = Product
        fields = ['id', 'product_id', 'name', 'collection', 'finishing', 'distributor']
0 голосов
/ 31 января 2020

Вдохновленный набором запросов @Arakkal Abu, я попробовал его с первой попытки.
FinishTypeSerializer добавлено в serializers.py: 

class FinishTypeSerializer(serializers.ModelSerializer):
    class Meta:
        model = FinishType
        fields = ('name',)

ProductGridSerializer в serializers.py: 

class ProductGridSerializer(serializers.ModelSerializer):
    name = serializers.SerializerMethodField(source='get_name')
    finishing = serializers.SerializerMethodField(source='get_finishing')
    distributor = serializers.ReadOnlyField(source='get_distributors')

    @staticmethod
    def get_name(obj):
        return [pa.name for pa in obj.productalias_set.all()]

    @staticmethod
    def get_finishing(product):
        return [pft.name for pft in FinishType.objects.filter(productfinishtype__product=product)]

    class Meta:
        model = Product
        fields = ['id', 'product_id', 'name', 'collection', 'finishing', 'distributor']

Выход JSON: 

{
     "id": 1,
     "product_id": "BQ1111",
     "name": [
           "White Stone"
     ],
     "collection": 1,
     "finishing": [
           "Polished",
           "Honed",
           "Carved"
     ],
     "distributor": 5
},
...