1
0
Fork 0
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
prezto/themes/twain/charge.py

38 lines
1.1 KiB

#!/usr/bin/env python
# coding=UTF-8
""" Prints a battery meter for use in a zsh theme """
import sys
""" Config """
output_length = 10
hide_when_charge_is_above = 95
blank_when_battery_full = True
filled_char, empty_char = u'', u''
""" Read battery charge """
def read_battery_percent (sp, regex):
import subprocess, re
process = subprocess.Popen(sp, stdout=subprocess.PIPE)
output = process.communicate()[0].strip()
return int(re.search(regex,output).group(1))
if sys.platform.startswith('darwin'):
""" Reads the battery charge as an integer between 0 and 100, using 'pmset -g batt' """
charge = read_battery_percent(["pmset", "-g", "batt"], "^.+([\d]+)\%")
else:
""" Reads the battery charge as an integer between 0 and 100, using 'acpi -b' """
charge = read_battery_percent(["acpi", "-b"], "^Battery 0: .+, ([\d]+)\%")
""" Output """
if charge < hide_when_charge_is_above:
filled = int(charge / output_length)
empty = output_length - filled
meter = (filled * filled_char + empty * empty_char).encode('utf-8')
color = 'green'if filled > 6 else 'yellow' if filled > 4 else 'red'
sys.stdout.write("%F{{{}}}{}%f".format(color, meter))