From d33bf1c4501a3d1312c4ed2d1c05e8a8d1b71363 Mon Sep 17 00:00:00 2001
From: xiaji <xiaji@example.com>
Date: Thu, 30 Apr 2026 14:16:45 +0800
Subject: [PATCH] Add battery count column to UPS host list

---
 .../ups_manager/templates/ups_manager/ups_list.html           | 4 +++-
 ups_management/ups_manager/views.py                           | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/ups_management/ups_manager/templates/ups_manager/ups_list.html b/ups_management/ups_manager/templates/ups_manager/ups_list.html
index 25d2e3c..b20d0ba 100644
--- a/ups_management/ups_manager/templates/ups_manager/ups_list.html
+++ b/ups_management/ups_manager/templates/ups_manager/ups_list.html
@@ -36,6 +36,7 @@
             <th>IP地址</th>
             <th>数量</th>
             <th>存放位置</th>
+            <th>电池数量</th>
             <th>上次维保时间</th>
             <th>联系人</th>
             <th>操作</th>
@@ -51,6 +52,7 @@
             <td>{{ ups.ip_address }}</td>
             <td>{{ ups.quantity }}</td>
             <td>{{ ups.location }}</td>
+            <td><span class="badge bg-success">{{ ups.battery_set.count }}</span></td>
             <td>{{ ups.last_maintenance_date|default:"-" }}</td>
             <td>{{ ups.contact|default:"-" }}</td>
             <td>
@@ -60,7 +62,7 @@
         </tr>
         {% empty %}
         <tr>
-            <td colspan="10" class="text-center text-muted">暂无UPS主机记录</td>
+            <td colspan="11" class="text-center text-muted">暂无UPS主机记录</td>
         </tr>
         {% endfor %}
     </tbody>
diff --git a/ups_management/ups_manager/views.py b/ups_management/ups_manager/views.py
index 793364b..d0ed918 100644
--- a/ups_management/ups_manager/views.py
+++ b/ups_management/ups_manager/views.py
@@ -78,7 +78,7 @@ class UPSHostListView(ListView):
     paginate_by = 10
 
     def get_queryset(self):
-        queryset = super().get_queryset()
+        queryset = super().get_queryset().prefetch_related('battery_set')
         brand = self.request.GET.get('brand')
         model = self.request.GET.get('model')
         location = self.request.GET.get('location')