1 files changed, 68 insertions, 0 deletions
diff --git a/py3status/modules/battery-level.py b/py3status/modules/battery-level.py
new file mode 100644
index 0000000..6c30556
--- /dev/null
+++ b/py3status/modules/battery-level.py
@@ -0,0 +1,68 @@
+# -*- coding: utf8 -*-
+
+from __future__ import division  # python2 compatibility
+from time import time
+
+import math
+import re
+import subprocess
+
+"""
+Module for displaying information about battery.
+
+Requires:
+    - the 'acpi' command line
+
+@author shadowprince
+@license Eclipse Public License
+"""
+
+CACHE_TIMEOUT = 30  # time to update battery
+HIDE_WHEN_FULL = False  # hide any information when battery is fully charged
+
+MODE = "bar"  # for primitive-one-char bar, or "text" for text percentage ouput
+
+BLOCKS = ["_", "▁", "▂", "▃", "▄", "▅", "▆", "▇", "█"]  # block for bar
+TEXT_FORMAT = "Battery: {}"  # text with "text" mode. percentage with % replaces {}
+
+CHARGING_CHARACTER = "⚡"
+
+# None means - get it from i3 config
+COLOR_BAD = None
+COLOR_CHARGING = "#FCE94F"
+COLOR_DEGRADED = None
+COLOR_GOOD = None
+
+
+class Py3status:
+    def battery_level(self, i3status_output_json, i3status_config):
+        response = {'name': 'battery-level'}
+
+        acpi = subprocess.check_output(["acpi"]).decode('utf-8')
+        proc = int(re.search(r"(\d+)%", acpi).group(1))
+
+        charging = bool(re.match(r".*Charging.*", acpi))
+        full = bool(re.match(r".*Unknown.*", acpi)) or bool(re.match(r".*Full.*", acpi))
+
+        if MODE == "bar":
+            character = BLOCKS[int(math.ceil(proc/100*(len(BLOCKS) - 1)))]
+        else:
+            character = TEXT_FORMAT.format(str(proc) + "%")
+
+        if proc < 30:
+            response['color'] = COLOR_DEGRADED if COLOR_DEGRADED else i3status_config['color_degraded']
+        if proc < 10:
+            response['color'] = COLOR_BAD if COLOR_BAD else i3status_config['color_bad']
+
+        if full:
+            response['color'] = COLOR_GOOD if COLOR_GOOD else i3status_config['color_good']
+            response['full_text'] = "" if HIDE_WHEN_FULL else BLOCKS[-1]
+        elif charging:
+            response['color'] = COLOR_CHARGING
+            response['full_text'] = CHARGING_CHARACTER
+        else:
+            response['full_text'] = character
+
+        response['cached_until'] = time() + CACHE_TIMEOUT
+
+        return (0, response)