From 43b33be51c71c1c4ed782ee26509596b98898e3b Mon Sep 17 00:00:00 2001
From: BowDown097 <42720004+BowDown097@users.noreply.github.com>
Date: Sun, 25 Sep 2022 09:50:53 -0700
Subject: [PATCH] battery_percentage fix (#437)

---
 auto_cpufreq/core.py | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/auto_cpufreq/core.py b/auto_cpufreq/core.py
index dffca15..f6d47da 100644
--- a/auto_cpufreq/core.py
+++ b/auto_cpufreq/core.py
@@ -259,7 +259,11 @@ def battery_percentage():
     """
     get batery percentage
     """
-    return round(psutil.sensors_battery().percent)
+    sensors_battery = psutil.sensors_battery()
+    if sensors_battery:
+        return round(sensors_battery.percent)
+    else:
+        return None
 
 
 def get_avail_gov():
@@ -998,11 +1002,14 @@ def set_autofreq():
     print("\n" + "-" * 28 + " CPU frequency scaling " + "-" * 28 + "\n")
 
     # determine which governor should be used
+    battery_percent = battery_percentage()
     if charging():
-        print("Battery is: charging (" + str(battery_percentage()) + "%)\n")
+        if battery_percent:
+            print("Battery is: charging (" + str(battery_percent) + "%)\n")
         set_performance()
     else:
-        print("Battery is: discharging (" + str(battery_percentage()) + "%)\n")
+        if battery_percent:
+            print("Battery is: discharging (" + str(battery_percent) + "%)\n")
         set_powersave()
 
 
@@ -1014,13 +1021,16 @@ def mon_autofreq():
     print("\n" + "-" * 28 + " CPU frequency scaling " + "-" * 28 + "\n")
 
     # determine which governor should be used
+    battery_percent = battery_percentage()
     if charging():
-        print("Battery is: charging (" + str(battery_percentage()) + "%)\n")
+        if battery_percent:
+            print("Battery is: charging (" + str(battery_percent) + "%)\n")
         get_current_gov()
         print(f'Suggesting use of "{get_avail_performance()}" governor')
         mon_performance()
     else:
-        print("Battery is: discharging (" + str(battery_percentage()) + "%)\n")
+        if battery_percent:
+            print("Battery is: discharging (" + str(battery_percent) + "%)\n")
         get_current_gov()
         print(f'Suggesting use of "{get_avail_powersave()}" governor')
         mon_powersave()