From a06e97194bb686dc7a91b3c19eb7acd438cac7dc Mon Sep 17 00:00:00 2001 From: Luca Bilke Date: Tue, 19 Jul 2022 15:50:04 +0200 Subject: [PATCH] fix battery indicator when no battery found --- .local/bin/battery | 28 +++++++++++++++------------- 1 file changed, 15 insertions(+), 13 deletions(-) diff --git a/.local/bin/battery b/.local/bin/battery index eab88f16..7ddb3c72 100755 --- a/.local/bin/battery +++ b/.local/bin/battery @@ -31,22 +31,24 @@ def get_battery(): # return ret + PLUGGED_ICONS[min(PLUGGED_ICONS.keys(), key=lambda x: abs(x - battery.percent))] # return ret + UNPLUGGED_ICONS[min(UNPLUGGED_ICONS.keys(), key=lambda x: abs(x - battery.percent))] battery = psutil.sensors_battery() - ret = str(round(battery.percent)) + "% " - if not battery: - return ret + UNKOWN_ICON - if battery.power_plugged: + try: + ret = str(round(battery.percent)) + "% " + if not battery: + return ret + UNKOWN_ICON + if battery.power_plugged: + return ( + ret + + PLUGGED_ICONS[ + min(PLUGGED_ICONS.keys(), key=lambda x: abs(x * 10 - battery.percent)) + ] + ) return ( ret - + PLUGGED_ICONS[ - min(PLUGGED_ICONS.keys(), key=lambda x: abs(x * 10 - battery.percent)) + + UNPLUGGED_ICONS[ + min(UNPLUGGED_ICONS.keys(), key=lambda x: abs(x * 10 - battery.percent)) ] ) - return ( - ret - + UNPLUGGED_ICONS[ - min(UNPLUGGED_ICONS.keys(), key=lambda x: abs(x * 10 - battery.percent)) - ] - ) - + except AttributeError: + return "" print(get_battery())