battery-widget: Simplify capacity computation

Increase cohesion of code that computes total charge and total capacity.

1 files changed, 1 insertions, 4 deletions

diff --git a/battery-widget/battery.lua b/battery-widget/battery.lua
index 452d7ef..935f680 100644
--- a/battery-widget/battery.lua
+++ b/battery-widget/battery.lua
@@ -135,10 +135,6 @@ local function worker(user_args)
         end
 
         local capacity = 0
-        for _, cap in ipairs(capacities) do
-            capacity = capacity + cap
-        end
-
         local charge = 0
         local status
         for i, batt in ipairs(battery_info) do
@@ -149,6 +145,7 @@ local function worker(user_args)
                 end
 
                 charge = charge + batt.charge * capacities[i]
+                capacity = capacity + capacities[i]
             end
         end
         charge = charge / capacity