From 70e3a4f2b0a852f446161a76b3a6fd65d6e142de Mon Sep 17 00:00:00 2001
From: huang <1724659546@qq.com>
Date: Thu, 23 Oct 2025 12:14:39 +0800
Subject: [PATCH] =?UTF-8?q?=E7=8C=AA=E7=BE=A4=E7=AE=A1=E7=90=86=E8=AE=A9?=
 =?UTF-8?q?=E7=94=A8=E6=88=B7=E9=80=89=E6=8B=A9=E7=8C=AA=E6=A0=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/api/pigBatch.js                      | 18 ++++++
 src/components/PigBatchList.vue          | 73 ++++++++++++++++++++++--
 src/views/pms/PigBatchManagementView.vue | 13 ++++-
 3 files changed, 98 insertions(+), 6 deletions(-)

diff --git a/src/api/pigBatch.js b/src/api/pigBatch.js
index 0c7a5dc4..92d9ca3c 100644
--- a/src/api/pigBatch.js
+++ b/src/api/pigBatch.js
@@ -191,3 +191,21 @@ export const recordDeathInBatch = (id, deathData) => {
 export const recordCullInBatch = (id, cullData) => {
   return http.post(`/api/v1/pig-batches/record-cull/${id}`, cullData);
 };
+
+// --- 新增的猪栏和猪舍API ---
+
+/**
+ * 获取所有猪栏的列表
+ * @returns {Promise<*>}
+ */
+export const getAllPens = () => {
+  return http.get('/api/v1/pens');
+};
+
+/**
+ * 获取所有猪舍的列表
+ * @returns {Promise<*>}
+ */
+export const getAllPigHouses = () => {
+  return http.get('/api/v1/pig-houses');
+};
diff --git a/src/components/PigBatchList.vue b/src/components/PigBatchList.vue
index 807f5800..d68574a8 100644
--- a/src/components/PigBatchList.vue
+++ b/src/components/PigBatchList.vue
@@ -8,7 +8,7 @@
           <span>初始数量: {{ batch.initial_count }}</span>
         </div>
         <div class="batch-actions">
-          <el-button size="small" type="primary" @click.stop="emitAddPen(batch)">增加猪栏</el-button>
+          <el-button size="small" type="primary" @click.stop="showAddPenDialog(batch)">增加猪栏</el-button>
           <el-button size="small" @click.stop="emitEditBatch(batch)">编辑</el-button>
           <el-button size="small" type="danger" @click.stop="emitDeleteBatch(batch)">删除</el-button>
         </div>
@@ -28,11 +28,30 @@
         </div>
       </div>
     </div>
+
+    <!-- 增加猪栏对话框 -->
+    <el-dialog title="选择猪栏" v-model="addPenDialogVisible" width="30%">
+      <el-select v-model="selectedPenId" placeholder="请选择猪栏" style="width: 100%;">
+        <el-option
+          v-for="pen in availablePens"
+          :key="pen.id"
+          :label="pen.label"
+          :value="pen.id">
+        </el-option>
+      </el-select>
+      <template #footer>
+        <span class="dialog-footer">
+          <el-button @click="addPenDialogVisible = false">取 消</el-button>
+          <el-button type="primary" @click="assignPen">确 定</el-button>
+        </span>
+      </template>
+    </el-dialog>
   </div>
 </template>
 
 <script>
 import PigPenInfoCard from './PigPenInfoCard.vue';
+import { getAllPens, getAllPigHouses } from '../api/pigBatch'; // Import new API functions
 
 export default {
   name: 'PigBatchList',
@@ -45,15 +64,59 @@ export default {
       required: true
     }
   },
-  emits: ['edit-batch', 'delete-batch', 'add-pen', 'edit-pen', 'delete-pen'],
+  emits: ['edit-batch', 'delete-batch', 'add-pen', 'edit-pen', 'delete-pen', 'assign-pen-to-batch'],
+  data() {
+    return {
+      addPenDialogVisible: false,
+      availablePens: [],
+      selectedPenId: null,
+      currentBatch: null // To store the batch for which we are adding a pen
+    };
+  },
   methods: {
     toggleExpand(batch) {
       batch.isExpanded = !batch.isExpanded;
     },
-    // 猪群操作
-    emitAddPen(batch) {
-      this.$emit('add-pen', batch);
+    async fetchAvailablePens() {
+      try {
+        const [pensResponse, housesResponse] = await Promise.all([
+          getAllPens(),
+          getAllPigHouses()
+        ]);
+
+        const pens = pensResponse.data; 
+        const houses = housesResponse.data;
+
+        // Create a map for quick lookup of house names by ID
+        const houseMap = new Map(houses.map(house => [house.id, house.name]));
+
+        this.availablePens = pens.map(pen => ({
+          id: pen.id,
+          label: `${pen.pen_number}(${houseMap.get(pen.house_id) || '未知猪舍'})` // Changed format here
+        }));
+      } catch (error) {
+        console.error("Error fetching pens or houses:", error);
+        this.$message.error("获取猪栏或猪舍信息失败");
+      }
     },
+    showAddPenDialog(batch) {
+      this.currentBatch = batch;
+      this.selectedPenId = null; // Reset selection
+      this.fetchAvailablePens();
+      this.addPenDialogVisible = true;
+    },
+    assignPen() {
+      if (this.selectedPenId && this.currentBatch) {
+        this.$emit('assign-pen-to-batch', {
+          batchId: this.currentBatch.id,
+          penId: this.selectedPenId
+        });
+        this.addPenDialogVisible = false;
+      } else {
+        this.$message.warning("请选择一个猪栏");
+      }
+    },
+    // 猪群操作
     emitEditBatch(batch) {
       this.$emit('edit-batch', batch);
     },
diff --git a/src/views/pms/PigBatchManagementView.vue b/src/views/pms/PigBatchManagementView.vue
index 77836f10..3fea6789 100644
--- a/src/views/pms/PigBatchManagementView.vue
+++ b/src/views/pms/PigBatchManagementView.vue
@@ -41,6 +41,7 @@
           @add-pen="handleAddPen"
           @edit-pen="handleEditPen"
           @delete-pen="handleDeletePen"
+          @assign-pen-to-batch="handleAssignPenToBatch"
         />
         <el-empty v-else description="暂无数据" />
       </div>
@@ -68,7 +69,7 @@
 </template>
 
 <script>
-import { getPigBatches, deletePigBatch } from '@/api/pigBatch.js';
+import { getPigBatches, deletePigBatch, assignPensToBatch } from '@/api/pigBatch.js'; // Import assignPensToBatch
 import { getPens, deletePen } from '@/api/pen.js';
 import { getPigHouses } from '@/api/pigHouse.js';
 import PigBatchList from '@/components/PigBatchList.vue';
@@ -253,6 +254,16 @@ export default {
           this.$message.success('猪栏添加成功');
         }
       }
+    },
+    async handleAssignPenToBatch({ batchId, penId }) {
+      try {
+        await assignPensToBatch(batchId, { pen_ids: [penId] });
+        this.$message.success('猪栏分配成功');
+        await this.loadData(); // Refresh data to show assigned pen
+      } catch (error) {
+        this.$message.error('分配猪栏失败: ' + (error.response?.data?.message || error.message || '未知错误'));
+        console.error('Failed to assign pen to batch:', error);
+      }
     }
   }
 }