refactor(battery): hoist raw-state mapping to constant

This commit is contained in:
sankalp committed 2026-08-02 13:37:09 +05:45
1 parent 57b9269736
commit e949bd4397
1 file changed
+10 -7
+10 -7
View File
@@ -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: