feat: added put/patch api to main oobject of business module

api_v1/views/business/purchase/views.py

@@ -27,6 +27,7 @@ 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)
+    quantity_of_rolls = serializers.SerializerMethodField(read_only=True)
 
     def get_total_amount(self, obj: business_models.PurchaseOrder):
         return obj.get_total_amount()
@@ -37,6 +38,9 @@ class PurchaseOrderSerializer(serializers.ModelSerializer):
     def get_total_quantity(self, obj: business_models.PurchaseOrder):
         return obj.get_total_quantity()
 
+    def get_quantity_of_rolls(self, obj: business_models.PurchaseOrder):
+        return [item.split_quantity_of_rolls() for item in obj.items.all()]
+
     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)
@@ -49,6 +53,7 @@ class PurchaseOrderSerializer(serializers.ModelSerializer):
             'total_amount', 'diff_quantity', 'total_quantity',
             'operator', 'operator_name', 'warehouse', 'warehouse_name',
             'status', 'remarks', 'created_at', 'updated_at', 'items',
+            'quantity_of_rolls',
         ]
         read_only_fields = ['id', 'created_at', 'updated_at', 'items', 'supplier_name', 'operator_name', 'warehouse_name']
 
@@ -135,6 +140,73 @@ class PurchaseOrderView(StockChangeViewMixin, views.APIView):
         )
 
 
+class PurchaseOrderDetailView(StockChangeViewMixin, views.APIView):
+    permission_classes = [IsAuthenticated]
+
+    def _get_order(self, request, pk: int):
+        if not self.check_employee_permission(request):
+            return None, self.permission_error_response('无权限访问')
+        merchant = request.user.employee.merchant
+        try:
+            order = business_models.PurchaseOrder.objects.select_related(
+                'supplier', 'operator', 'warehouse'
+            ).prefetch_related('items').get(id=pk, merchant=merchant)
+            return order, None
+        except business_models.PurchaseOrder.DoesNotExist:
+            return None, self.not_found_response('采购单不存在')
+
+    def put(self, request, pk: int):
+        return self._update(request, pk)
+
+    def patch(self, request, pk: int):
+        return self._update(request, pk)
+
+    def _update(self, request, pk: int):
+        purchase_order, error_response = self._get_order(request, pk)
+        if error_response:
+            return error_response
+
+        merchant = request.user.employee.merchant
+        data = request.data or {}
+
+        supplier = purchase_order.supplier
+        supplier_id = data.get('supplier')
+        if supplier_id:
+            try:
+                supplier = basic_models.Supplier.objects.get(id=supplier_id, merchant=merchant)
+            except basic_models.Supplier.DoesNotExist:
+                return Response({'error': f'供应商 {supplier_id} 不存在'}, status=status.HTTP_400_BAD_REQUEST)
+
+        warehouse = purchase_order.warehouse
+        warehouse_id = data.get('warehouse_id') or data.get('warehouse')
+        if warehouse_id:
+            try:
+                warehouse = basic_models.WareHouse.objects.get(id=warehouse_id, merchant=merchant)
+            except basic_models.WareHouse.DoesNotExist:
+                return Response({'error': f'仓库 {warehouse_id} 不存在'}, status=status.HTTP_400_BAD_REQUEST)
+
+        order_date = data.get('order_date') or purchase_order.purchase_date
+        items = data.get('items')
+        if not isinstance(items, list) or not items:
+            return Response({'error': 'items 需要为非空数组'}, status=status.HTTP_400_BAD_REQUEST)
+        remarks = data.get('remarks', purchase_order.remarks)
+
+        try:
+            updated_order = business_services.update_purchase_order(
+                purchase_order=purchase_order,
+                supplier=supplier,
+                order_date=order_date,
+                warehouse=warehouse,
+                items=items,
+                remarks=remarks,
+            )
+        except ValueError as exc:
+            return Response({'error': str(exc)}, status=status.HTTP_400_BAD_REQUEST)
+
+        serializer = PurchaseOrderSerializer(updated_order)
+        return Response(serializer.data, status=status.HTTP_200_OK)
+
+
 class PurchaseOrderReviewSerializer(serializers.Serializer):
     action = serializers.ChoiceField(choices=[('approve', '审批通过'), ('cancel', '作废')])