diff options
Diffstat (limited to '')
| -rw-r--r-- | xpybar/config/mybattery.py | 163 |
1 files changed, 163 insertions, 0 deletions
diff --git a/xpybar/config/mybattery.py b/xpybar/config/mybattery.py new file mode 100644 index 0000000..3a8e167 --- /dev/null +++ b/xpybar/config/mybattery.py @@ -0,0 +1,163 @@ +# -*- python -*- +from plugins.powersupply import PowerSupply + +from common import * + +class MyBattery(Entry): + COMPACT = 0 + NORMAL = 1 + DETAILED = 2 + + def __init__(self, *args, show_all = False, show_battery = 0, details = 1, batteries = None, **kwargs): + self.show_all = show_all + self.show_battery = show_battery + self.supply_filter = batteries + self.details = details + self.supplies = None + self.batteries = None + self.reload() + Entry.__init__(self, *args, **kwargs) + + def reload(self): + name = None + newsupplies = {} + newbatteries = [] + if self.batteries is not None: + if isinstance(self.batteries, int): + name = self.batteries[self.show_battery] + else: + name = self.show_battery + for supply in PowerSupply.supplies(): + if self.supply_filter is not None: + if supply not in self.supply_filter: + continue + supply = PowerSupply(supply) + if supply.type != 'Battery': + continue + newsupplies[supply.name] = supply + newbatteries.append(supply.name) + if self.supply_filter is not None: + newbatteries = [b for b in self.supply_filter if b in newsupplies] + if name is None: + index = 0 + elif isinstance(name, int): + index = name % len(newbatteries) if len(newbatteries) > 0 else 0 + elif name in self.batteries: + index = self.batteries.index(index) + else: + index = 0 + self.supplies, self.batteries, self.show_battery = newsupplies, newbatteries, index + + def action(self, col, button, x, y): + if button == LEFT_BUTTON: + self.show_all = not self.show_all + self.invalidate() + elif button == MIDDLE_BUTTON: + self.details += 1 + self.details %= 3 + self.invalidate() + elif button == RIGHT_BUTTON: + self.reload() + self.invalidate() + elif button == SCROLL_UP: + if len(self.batteries) > 0: + self.show_battery += 1 + self.show_battery %= len(self.batteries) + self.invalidate() + elif button == SCROLL_DOWN: + if len(self.batteries) > 0: + self.show_battery -= 1 + if self.show_battery < 0: + self.show_battery += len(self.batteries) + self.invalidate() + + def colourise(self, text): + try: + value = text.replace(',', '.') + value = float(value) if '.' in value else int(value) + colour = '39' + if value < 80: colour = '32' + if value < 50: colour = '33' + if value < 15: colour = '31' + if value < 5: colour = '7;31' + return '\033[%sm%s\033[0m' % (colour, text) + except: + return text + + def battery(self, bat): + bat = self.supplies[bat] + name = bat.name[:1].upper() + bat.name[1:].lower() + if bat.is_online(): + try: + charge = (bat.get_charge(), bat.get_charge_full(), bat.charge_full_design) + current = bat.get_current() + status = bat.get_status().lower() + left = None + if charge[0] == charge[1]: + capacity = '100' + else: + capacity = '%.1f' % (charge[0] / charge[1] * 100) + capacity = self.colourise(capacity) + + if current > 0: + if status == 'charging': + left = (charge[1] - charge[0]) / current + elif status == 'discharging': + left = charge[0] / current + if left is not None: + hours = int(left) + minutes = int((left - hours) * 60 + 0.5) + if minutes >= 60: + minutes -= 60 + hours += 1 + left = '%ih%02im until %s' % (hours, minutes, status.replace('ing', 'ed')) + + if self.details == MyBattery.COMPACT: + if status == 'full': + return 'full' + elif status == 'charging': + status = '+' + elif status == 'discharging': + status = '-' + else: + status = '?' + if left is None: + return '%s%%%s' % (capacity, status) + else: + return '%s%%%s %s' % (capacity, status, left) + elif self.details == MyBattery.NORMAL: + if left is None: + return '%s: %s%% %s' % (name, capacity, status) + else: + return '%s: %s%% %s %s' % (name, capacity, status, left) + else: + if charge[1] == charge[2]: + health = '100' + else: + health = '%.1f' % (charge[1] / charge[2] * 100) + health = self.colourise(health) + volt = bat.get_voltage() / bat.voltage_min_design + if volt < 1: + volt = '\033[31m%.1f\033[0m' % volt + elif volt < 1.05: + volt = '\033[33m%.1f\033[0m' % volt + else: + volt = '%.1f' % volt + if left is None: + return '%s: %s%% %s, %s%% healthy, %sx voltage' % (name, capacity, status, health, volt) + else: + return '%s: %s%% %s %s, %s%% healthy, %sx voltage' % (name, capacity, status, left, health, volt) + except Exception as e: + return (repr(e)) + return ('%s: ???' % name) if self.details > 0 else '???' + else: + return ('%s: offline' % name) if self.details > 0 else 'offline' + + def function(self): + if len(self.batteries) == 0: + return 'no batteries available' + + if self.show_all: + return SEPARATOR.join(self.battery(bat) for bat in self.batteries) + else: + return self.battery(self.batteries[self.show_battery]) |