首页 > 解决方案 > 如何在Django rest框架中检查数据是否存在

问题描述

我有一个 CartModel、CartItem 模型。我在创建购物车时尝试做,当购物车被创建时,我再次添加与之前输入相同的数据,它会创建另一个购物车。如果购物车存在于 CartItem 表中,我试图这样做,它必须显示响应“购物车已经存在”。但它正在创建另一个购物车。我搜索了很多并使用了代码,但它不会运行。如果有人能够给出答案,请帮助我。

视图.py *

class CartTestViewSet(viewsets.ModelViewSet):
    serializer_class = CartItemSerializer
    permission_classes = (IsAuthenticated,)
    
    def get_queryset(self):
        user = self.request.user
        if user.is_authenticated:
            if user is not None:
                if user.is_active and user.is_superuser or user.is_Customer:
                    return CartItem.objects.all()
                raise PermissionDenied()
            raise PermissionDenied()
        raise PermissionDenied()

    def post(self, request):       
        cart = Cart.objects.filter(user=request.user, cart=request.data, service=request.data, defects=request.data)
        if cart.exists():   
            print('cart already exists')
        else:
            print('cart is created')

序列化程序.py

from rest_framework import serializers
from .models import CartItem, CartModel
from rest_framework.response import Response

class CartItemSerializer(serializers.ModelSerializer):
    def get_total(self, obj):
        return obj.get_total
    get_total = serializers.IntegerField(read_only=True, required=False)
    # id = serializers.IntegerField(read_only=False)
    class Meta:
        model = CartItem
        fields = "__all__"
        extra_fields = ['get_total']
        read_only = ['get_total']

class CartModelSerializer(serializers.ModelSerializer):
    class Meta:
        model = CartModel
        fields = "__all__"

模型.py

from django.db import models
from accounts.models import User, SubCategory
from category.models import Defects
from django.utils import timezone
# Create your models here.

class CartModel(models.Model):
    user = models.ForeignKey('accounts.User', on_delete=models.CASCADE)
    status_choice = [
        ('1', 'open'),
        ('2', 'closed')
    ]
    status = models.CharField(max_length=2, choices=status_choice, default=1)
    
    def __str__(self):
        return self.user.username
    
class CartItem(models.Model):
    cart = models.ForeignKey('CartModel', on_delete=models.CASCADE)
    user = models.ForeignKey('accounts.User', on_delete=models.CASCADE)
    service = models.ForeignKey('accounts.SubCategory', on_delete=models.CASCADE)
    defects = models.ForeignKey('category.Defects', on_delete=models.CASCADE)
    quantity = models.IntegerField(default=1)
    price = models.IntegerField()
    created_on = models.DateTimeField(auto_now_add=True)
    updated_on = models.DateTimeField(auto_now_add=True)


    def __str__(self):
        total = int(self.quantity) * int(self.price)
        return str(total)

    
    def get_total(self):
        total = int(self.quantity )* int(self.price)
        return total

标签: pythondjangoapidjango-rest-frameworkdjango-serializer

解决方案


您可以unique=True在 CartItem 模型中使用。如果您尝试使用此保存重复值,django 将引发错误。

cart = models.ForeignKey('CartModel', on_delete=models.CASCADE, unique=True)

推荐阅读