From e949bd439768d5b554de3c3d83d59195aa601e46 Mon Sep 17 00:00:00 2001
From: S4NKALP <sankalptharu50028@gmail.com>
Date: Sun, 2 Aug 2026 13:37:09 +0545
Subject: [PATCH] refactor(battery): hoist raw-state mapping to constant

---
 src/services/battery.py | 17 ++++++++++-------
 1 file changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/services/battery.py b/src/services/battery.py
index f81a8ee..312c1fc 100644
--- a/src/services/battery.py
+++ b/src/services/battery.py
@@ -133,14 +133,17 @@ def _level_icon(capacity: int) -> str:
     return "battery-level-08-symbolic"
 
 
+_STATE_FROM_RAW = {
+    DeviceState.CHARGING: BatteryStatus.CHARGING,
+    DeviceState.DISCHARGING: BatteryStatus.DISCHARGING,
+    DeviceState.FULLY_CHARGED: BatteryStatus.FULL,
+    DeviceState.PENDING_CHARGE: BatteryStatus.NOT_CHARGING,
+    DeviceState.PENDING_DISCHARGE: BatteryStatus.NOT_CHARGING,
+}
+
+
 def _state_from_raw(state: int) -> BatteryStatus:
-    return {
-        DeviceState.CHARGING: BatteryStatus.CHARGING,
-        DeviceState.DISCHARGING: BatteryStatus.DISCHARGING,
-        DeviceState.FULLY_CHARGED: BatteryStatus.FULL,
-        DeviceState.PENDING_CHARGE: BatteryStatus.NOT_CHARGING,
-        DeviceState.PENDING_DISCHARGE: BatteryStatus.NOT_CHARGING,
-    }.get(state, BatteryStatus.UNKNOWN)
+    return _STATE_FROM_RAW.get(state, BatteryStatus.UNKNOWN)
 
 
 def _device_kind_from_u32(value: int) -> UpDeviceKind: