如何在Django REST中自动更新数据库并实现订单状态查询接口？

问题一：如何在Django REST框架中自动更新数据库？ #

实现数据库自动更新，常用三种方案，按需选择：

1. Django信号（推荐，解耦性高） #

通过Django的post_save信号监听模型的保存事件，在关联对象（如订单商品）发生变化时，自动更新目标对象（如订单状态）。示例代码已在模型层实现，当OrderItems创建或更新后，自动触发订单状态的计算与更新。

2. 重写模型save()方法 #

在需要触发更新的模型（如OrderItems）中重写save()方法，在保存时主动调用关联对象的更新逻辑。示例：

class OrderItems(models.Model):
    # ... 其他字段
    def save(self, *args, **kwargs):
        super().save(*args, **kwargs)
        self.OrderNumber.update_order_status()

3. 视图层手动处理 #

在视图的perform_create或perform_update方法中，手动调用更新逻辑。这种方式耦合性较高，适合特定场景：

class OrderApiViewSet(ModelViewSet):
    # ... 其他代码
    def perform_create(self, serializer):
        instance = serializer.save()
        instance.OrderNumber.update_order_status()

问题二：订单状态功能实现（多商品独立状态+整体状态计算） #

核心逻辑 #

• 订单整体状态规则：存在任意PENDING（待发货）商品时，订单状态为PENDING；所有商品均为SHIPPED（已发货）时，订单状态为SHIPPED。
• 通过模型方法+信号自动维护订单状态，同时在序列化层返回订单及关联商品信息。

修改后完整代码 #

models.py

from django.db import models
from django.db.models.signals import post_save
from django.dispatch import receiver

# 公共状态选项，避免重复定义
ORDER_STATUS_CHOICES = (
    ('SHIPPED', '已发货'),
    ('PENDING', '待发货'),
    ('CANCELLED', '已取消'),
)

class OrderStatus(models.Model):
    OrderNumber = models.CharField(max_length=255, unique=True, verbose_name='订单ID')
    status = models.CharField(max_length=255, choices=ORDER_STATUS_CHOICES, default='PENDING', verbose_name='订单状态')

    def __str__(self):
        return self.OrderNumber

    def update_order_status(self):
        """计算并更新订单整体状态"""
        items = self.items.all()
        if not items.exists():
            self.status = 'PENDING'
        elif all(item.status == 'SHIPPED' for item in items):
            self.status = 'SHIPPED'
        else:
            # 只要存在待发货商品，订单状态保持待发货
            has_pending = any(item.status == 'PENDING' for item in items)
            self.status = 'PENDING' if has_pending else self.status
        self.save()

class OrderItems(models.Model):
    itemName = models.CharField(max_length=255, verbose_name='商品名称')
    OrderNumber = models.ForeignKey(OrderStatus, related_name='items', on_delete=models.CASCADE, verbose_name='关联订单')
    status = models.CharField(max_length=255, choices=ORDER_STATUS_CHOICES, default='PENDING', verbose_name='商品状态')

    def __str__(self):
        return f"{self.itemName} - {self.status}"

# 信号监听：商品状态变化时自动更新订单状态
@receiver(post_save, sender=OrderItems)
def update_order_on_item_change(sender, instance, **kwargs):
    instance.OrderNumber.update_order_status()

serializers.py

from rest_framework import serializers
from OrderStatus.models import OrderItems, OrderStatus

class OrderItemSerializer(serializers.ModelSerializer):
    """商品序列化器，返回商品详情"""
    class Meta:
        model = OrderItems
        fields = ['id', 'itemName', 'status']

class NumberOrderSerializer(serializers.ModelSerializer):
    """订单序列化器，嵌套返回商品列表与订单状态"""
    items = OrderItemSerializer(many=True, read_only=True)

    class Meta:
        model = OrderStatus
        fields = ['OrderNumber', 'status', 'items']

class OrderRegisterSerializer(serializers.ModelSerializer):
    """商品创建/更新序列化器"""
    class Meta:
        model = OrderItems
        fields = ['itemName', 'OrderNumber', 'status']

views.py

from rest_framework.viewsets import ModelViewSet
from OrderStatus.api.serializers import OrderRegisterSerializer, NumberOrderSerializer
from OrderStatus.models import OrderItems, OrderStatus

class OrderApiViewSet(ModelViewSet):
    serializer_class = OrderRegisterSerializer
    queryset = OrderItems.objects.all()

class OrderNumberApiviewSet(ModelViewSet):
    serializer_class = NumberOrderSerializer
    queryset = OrderStatus.objects.all()

功能说明 #

1. 自动状态维护：当通过API或后台修改商品状态时，信号会自动触发订单状态的计算与更新，无需手动干预。
2. 接口返回内容：访问订单接口（OrderNumberApiviewSet）时，会返回订单ID、整体状态，以及该订单下所有商品的ID、名称和各自状态。
3. 扩展性：如果后续需要添加其他状态规则（如包含取消商品时订单设为已取消），只需修改update_order_status方法中的逻辑即可。

内容的提问来源于stack exchange，提问作者Miguel Sanchez