feat: purchase_order approve and calcel

api_v1/views/purchase_order.py

@@ -1,7 +1,6 @@
 from rest_framework import status, views, serializers, pagination
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
-
 from basic_info import models as basic_models
 from business import services as business_services
 from business import models as business_models
@@ -14,12 +13,23 @@ class PurchaseOrderItemSerializer(serializers.ModelSerializer):
         fields = [
             'id', 'product', 'price', 'color', 'quantity', 'unit',
             'empty_diff_percent', 'quantity_of_rolls', 'num_of_rolls',
-            'batch_number', 'remarks', 'created_at', 'updated_at',
+            'batch_number', 'remarks', 'created_at', 'updated_at', 'spec',
         ]
-        read_only_fields = ['id', 'created_at', 'updated_at']
+        read_only_fields = ['id', 'created_at', 'updated_at', 'total_amount', 'diff_quantity', 'real_quantity']
 
 
 class PurchaseOrderSerializer(serializers.ModelSerializer):
+    total_amount = serializers.SerializerMethodField(read_only=True)
+    diff_quantity = serializers.SerializerMethodField(read_only=True)
+    total_quantity = serializers.SerializerMethodField(read_only=True)
+
+    def get_total_amount(self, obj: business_models.PurchaseOrder):
+        return obj.get_total_amount()
+    def get_diff_quantity(self, obj: business_models.PurchaseOrder):
+        return obj.get_total_diff_quantity()
+    def get_total_quantity(self, obj: business_models.PurchaseOrder):
+        return obj.get_total_quantity()
+
     supplier_name = serializers.CharField(source='supplier.name', read_only=True)
     operator_name = serializers.CharField(source='operator.name', read_only=True)
     warehouse_name = serializers.CharField(source='warehouse.name', read_only=True)
@@ -29,6 +39,7 @@ class PurchaseOrderSerializer(serializers.ModelSerializer):
         model = business_models.PurchaseOrder
         fields = [
             'id', 'supplier', 'supplier_name', 'purchase_date', 'kind',
+            'total_amount', 'diff_quantity', 'total_quantity',
             'operator', 'operator_name', 'warehouse', 'warehouse_name',
             'status', 'remarks', 'created_at', 'updated_at', 'items',
         ]
@@ -108,8 +119,53 @@ class PurchaseOrderView(StockChangeViewMixin, views.APIView):
         return Response(
             {
                 'id': purchase_order.id,
-                'message': '采购单创建成功，入库任务已排队',
+                'status': purchase_order.status,
+                'message': '采购单创建成功，等待审批',
             },
             status=status.HTTP_201_CREATED,
         )
 
+
+class PurchaseOrderReviewSerializer(serializers.Serializer):
+    action = serializers.ChoiceField(choices=[('approve', '审批通过'), ('cancel', '作废')])
+
+
+class PurchaseOrderReviewView(StockChangeViewMixin, views.APIView):
+    """采购单审批 / 作废"""
+
+    permission_classes = [IsAuthenticated]
+    ACTION_STATUS_MAP = {
+        'approve': business_models.PurchaseOrderStatusEnum.APPROVED,
+        'cancel': business_models.PurchaseOrderStatusEnum.CANCELLED,
+    }
+
+    def post(self, request, pk: int):
+        if not self.check_employee_permission(request):
+            return self.permission_error_response('无权限访问')
+
+        merchant = request.user.employee.merchant
+        try:
+            purchase_order = business_models.PurchaseOrder.objects.select_related(
+                'supplier', 'operator', 'warehouse'
+            ).prefetch_related('items').get(id=pk, merchant=merchant)
+        except business_models.PurchaseOrder.DoesNotExist:
+            return self.not_found_response('采购单不存在')
+
+        serializer = PurchaseOrderReviewSerializer(data=request.data or {})
+        serializer.is_valid(raise_exception=True)
+        action = serializer.validated_data['action']
+        target_status = self.ACTION_STATUS_MAP[action]
+
+        try:
+            business_services.review_purchase_order(
+                purchase_order=purchase_order,
+                target_status=target_status,
+                reviewed_by=request.user,
+            )
+        except ValueError as exc:
+            return Response({'error': str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+
+        refreshed_order = business_models.PurchaseOrder.objects.select_related(
+            'supplier', 'operator', 'warehouse'
+        ).prefetch_related('items').get(id=purchase_order.id)
+        return Response(PurchaseOrderSerializer(refreshed_order).data, status=status.HTTP_200_OK)