Как сравнить два поля сериализатора и показать, какое из них больше в django rest

У меня есть product serializer, который возвращает category_offer_price & product_offer_price, прежде чем получить этот ответ, я хочу сравнить обе цены и вернуть только ту цену, которая выше.

#Serilaizer.py

class ProductSerializer(ModelSerializer):
    category = CategorySerializer()
    product_offer_price = SerializerMethodField()
    category_offer_price = SerializerMethodField()

    class Meta:
        model = Products
        fields = [
            "id",
            "product_name",
            "slug",
            "category",
            "description",
            "category_offer_price",
            "product_offer_price",
            "base_price",
            "stock",
            "is_available",
            "created_date",
            "images",
            "images_two",
            "images_three",
        ]

    def get_product_offer_price(self, obj):
        try:
            product_offer = ProductOffer.objects.get(product=obj)
            if product_offer.is_active:
                offer_price = product_offer.product_offer_price()
                return offer_price
        except Exception:
            pass
        return None

    def get_category_offer_price(self, obj):
        try:
            category_offer = CategoryOffer.objects.get(category=obj.category)
            if category_offer.is_active:
                offer_price = category_offer.category_offer_price(obj)
                return offer_price
        except Exception:
            pass
        return None

#Models.py

class Products(models.Model):
    category = models.ForeignKey(Category, on_delete=models.CASCADE)
    product_name = models.CharField(max_length=50, unique=True)
    slug = models.SlugField(max_length=100, unique=True)
    description = models.TextField(max_length=500)
    base_price = models.IntegerField()
    images = models.ImageField(upload_to="photos/products")
    images_two = models.ImageField(upload_to="photos/products")
    images_three = models.ImageField(upload_to="photos/products")
    stock = models.IntegerField()
    is_available = models.BooleanField(default=True)
    created_date = models.DateTimeField(auto_now_add=True)
    modified_date = models.DateTimeField(auto_now=True)

    class Meta:
        verbose_name_plural = "Products"

    def __str__(self):
        return self.product_name

Я хотел бы узнать, возможно ли сравнить поля сериализатора в классе сериализатора?

Вы можете перейти к одному методу, для валидации вашего поля. Также замените метод try:except на метод get-object-or-404, а поля сериализатора на значение all, так как вы используете все, чтобы код был намного чище.

from django.shortcuts import get_object_or_404

class ProductSerializer(ModelSerializer):
    category = CategorySerializer()
    price = SerializerMethodField()

    class Meta:
        model = Products
        fields = '__all__'

    def get_price(self, obj):
        product_offer = get_object_or_404(ProductOffer, product=obj)
        category_offer = get_object_or_404(CategoryOffer, category=obj.category)

        if product_offer.is_active and category_offer.is_active:
            if product_offer.product_offer_price() > category_offer.category_offer_price(obj):
                    return product_offer.product_offer_price()
            else:
                    return category_offer.category_offer_price(obj)

        elif product_offer.is_active and not category_offer.is_active:
            return product_offer.product_offer_price()

        elif category_offer.is_active and not product_offer.is_active:
            return category_offer.category_offer_price(obj)

EDIT: Как вы видите, я использовал классические if/else в этом решении, хотя начиная с Python 3.10 вы можете использовать Match case statement для подстановки цепочки этих условий.

Вы можете отменить to_representation()

Пример:

class ProductSerializer(ModelSerializer):
    category = CategorySerializer()
    product_offer_price = SerializerMethodField()
    category_offer_price = SerializerMethodField()

    ...

    ...
    def to_representation(self, instance):
        data = super().to_representation(instance)
        # access required fields like this
        product_offer_price = data['product_offer_price']
        category_offer_price = data['category_offer_price']

        # do calculations here and returning the desired field as `calculated_price`
        if category_offer_price > product_offer_price:
            data['calculated_price'] = category_offer_price
        else:
            data['calculated_price'] = product_offer_price
        return data

Не уверен, что это то, что вы хотите, но вы можете использовать поле типа SerializerMethodField, которое позволит вам добавить вычисляемое поле, которое вы можете назвать category_offer_higher_price. Его значение вычисляется функцией, которая возвращает самую высокую цену. Смотрите следующую ссылку : https://www.django-rest-framework.org/api-guide/fields/#serializermethodfield

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