Fix battery count display - show total quantity instead of record count

2026-04-30 14:23:18 +08:00
parent d33bf1c450
commit 795c84ce16
2 changed files with 5 additions and 1 deletions
--- a/ups_management/ups_manager/models.py
+++ b/ups_management/ups_manager/models.py
@@ -44,6 +44,10 @@ class UPSHost(models.Model):
    created_at = models.DateTimeField(auto_now_add=True, verbose_name='创建时间')
    updated_at = models.DateTimeField(auto_now=True, verbose_name='更新时间')

+    @property
+    def total_battery_count(self):
+        return sum(battery.quantity for battery in self.battery_set.all())
+
    def __str__(self):
        if self.ip_address and not self.ip_address.startswith('-'):
            return f'{self.brand} {self.model} - {self.ip_address}'