-
-
Save benwah/90d9cd1ef96e6c3917e0e0f34c772828 to your computer and use it in GitHub Desktop.
dwstat.py - my dwm status bar script
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/usr/bin/env python | |
| from threading import Thread,Lock,Event,Timer | |
| import os | |
| import sys | |
| import signal | |
| import subprocess | |
| import math | |
| import psutil | |
| import re | |
| import time | |
| mutex = Lock() | |
| thread_data = [] | |
| class UpdaterThread(Thread): | |
| def __init__(self, id, sleep, callback): | |
| Thread.__init__(self) | |
| self.stopped = Event() | |
| self.id = id | |
| self.sleeptime = sleep | |
| self.callback = callback | |
| def stop(self): | |
| self.stopped.set() | |
| def run(self): | |
| thread_data[self.id] = self.callback() | |
| with mutex: | |
| update() | |
| while not self.stopped.is_set(): | |
| flagged = self.stopped.wait( | |
| self.sleeptime | |
| ) | |
| if not flagged: | |
| thread_data[self.id] = self.callback() | |
| with mutex: | |
| update() | |
| def update(): | |
| subprocess.call([ | |
| "xsetroot", | |
| "-name", | |
| " | ".join(thread_data) | |
| ]) | |
| def start(conf): | |
| threads = [] | |
| for i in range(len(conf)): | |
| thread_data.append("...") | |
| t = UpdaterThread( | |
| i, | |
| conf[i]['sleep'], | |
| conf[i]['callback'] | |
| ) | |
| t.setDaemon(True) | |
| threads.append(t) | |
| for t in threads: | |
| t.start() | |
| try: | |
| signal.pause() | |
| except (KeyboardInterrupt, SystemExit): | |
| print("stopping threads") | |
| for t in threads: | |
| t.stop() | |
| for t in threads: | |
| t.join() | |
| print("bye!") | |
| return 0 | |
| #### vvv configuration vvv #### | |
| def datetime(): | |
| return "".join([' ',time.strftime("%m-%d-%y - %H:%M:%S"),' ']) | |
| def wifi(): | |
| dev = "wlp2s0" | |
| wifi_info = open("/proc/net/wireless").read() | |
| signal_strength = 0 | |
| try: | |
| essid = re.sub( | |
| r'^.*"([^"]*).*"', | |
| r'\1', | |
| str(subprocess.check_output( | |
| ["iwgetid"], | |
| shell=False | |
| ).strip())) | |
| except: | |
| essid = "(x)" | |
| try: | |
| signal_strength = int(int(re.findall(r'{}:\s*\S*\s*(\d*)'.format(dev), wifi_info)[0])/70.0*100) | |
| except: | |
| signal_strength = 0 | |
| output = ['wi: '] | |
| output.append('({}) '.format(essid)) | |
| output.append('{}'.format(percent(signal_strength))) | |
| return "".join(output) | |
| def battery(): | |
| acpath = "/sys/class/power_supply/AC0" | |
| batpath = "/sys/class/power_supply/BAT0" | |
| ac_online = int(open(acpath+'/online','r').read().strip()) | |
| max = int(open(batpath+'/energy_full','r').read().strip()) | |
| cur = int(open(batpath+'/energy_now','r').read().strip()) | |
| pct = int(100*float(cur)/float(max)) | |
| output = ['bat: '] | |
| output.append("{}".format(percent(pct))) | |
| output.append("(") | |
| if ac_online: | |
| if pct == 100: | |
| output.append("+") | |
| else: | |
| output.append("~") | |
| else: | |
| output.append("-") | |
| output.append(")") | |
| return "".join(output) | |
| def cpu(): | |
| cpus = psutil.cpu_percent(interval=.5, percpu=True) | |
| mhz = re.search( | |
| "cpu MHz\s*:\s*(\S*)", | |
| open('/proc/cpuinfo','r' | |
| ).read()).groups()[0].split('.')[0] | |
| output = [' cpu: ']; | |
| output.append('{}Mz'.format(mhz)) | |
| output.append('(') | |
| cpu_data = " ".join(map(lambda x: percent(int(x)), cpus)) | |
| output.append(cpu_data) | |
| # for c in cpus: | |
| # output.append('{}'.format(percent(int(c)))) | |
| # #output.append("{0: >3}% ".format(int(c))) | |
| output.append(')') | |
| return "".join(output) | |
| def mem(): | |
| minfo = psutil.virtual_memory() | |
| output = ['mem:'] | |
| output.append("{0: >3}%".format(int(minfo.percent))) | |
| return "".join(output) | |
| def vol(): | |
| mixerinfo = subprocess. \ | |
| check_output(['amixer', 'get', 'Master'], shell=False). \ | |
| decode('utf-8'). \ | |
| split('\n')[-2] | |
| master_vol = int(re.search(r'\[(\d*)%\]', mixerinfo).groups()[0]) | |
| muted = re.search(r'\[(on|off)\]', mixerinfo).groups()[0] | |
| output = ['vol: '] | |
| output.append("{}".format(percent(master_vol))) | |
| if muted == 'off': | |
| output.append('M') | |
| return "".join(output) | |
| def percent(pct=0): | |
| return f"{pct}%" | |
| if __name__ == '__main__': | |
| conf = [ | |
| { | |
| "sleep":2, | |
| "callback":cpu, | |
| }, | |
| { | |
| "sleep":10, | |
| "callback":mem, | |
| }, | |
| { | |
| "sleep":3, | |
| "callback":wifi, | |
| }, | |
| { | |
| "sleep":1, | |
| "callback":vol, | |
| }, | |
| { | |
| "sleep":10, | |
| "callback":battery, | |
| }, | |
| { | |
| "sleep":1, | |
| "callback":datetime, | |
| }, | |
| ] | |
| sys.exit(start(conf)) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment