From c69da4c4f55a9f942f9729aa9942f2938397d6dd Mon Sep 17 00:00:00 2001 From: Valentin Haenel Date: Sun, 12 Aug 2012 17:57:48 +0200 Subject: [PATCH] contrib: python version of the wrapper --- contrib/wrapper.py | 71 ++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 71 insertions(+) create mode 100755 contrib/wrapper.py diff --git a/contrib/wrapper.py b/contrib/wrapper.py new file mode 100755 index 0000000..ccc292b --- /dev/null +++ b/contrib/wrapper.py @@ -0,0 +1,71 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +# This script is a simple wrapper which prefixes each i3status line with custom +# information. It is a python reimplementation of: +# http://code.stapelberg.de/git/i3status/tree/contrib/wrapper.pl +# +# To use it, ensure your ~/.i3status.conf contains this line: +# output_format = "i3bar" +# in the 'general' section. +# Then, in your ~/.i3/config, use: +# status_command i3status | ~/i3status/contrib/wrapper.py +# In the 'bar' section. +# +# In its current version it will display the cpu frequency governor, but you +# are free to change it to display whatever you like, see the comment in the +# source code below. +# +# © 2012 Valentin Haenel +# +# This program is free software. It comes without any warranty, to the extent +# permitted by applicable law. You can redistribute it and/or modify it under +# the terms of the Do What The Fuck You Want To Public License (WTFPL), Version +# 2, as published by Sam Hocevar. See http://sam.zoy.org/wtfpl/COPYING for more +# details. + +import sys +import json + +def get_governor(): + """ Get the current governor for cpu0, assuming all CPUs use the same. """ + with open('/sys/devices/system/cpu/cpu0/cpufreq/scaling_governor') as fp: + return fp.readlines()[0].strip() + +def print_line(message): + """ Non-buffered printing to stdout. """ + sys.stdout.write(message + '\n') + sys.stdout.flush() + +def read_line(): + """ Interrupted respecting reader for stdin. """ + # try reading a line, removing any extra whitespace + try: + line = sys.stdin.readline().strip() + # i3status sends EOF, or an empty line + if not line: + sys.exit(3) + return line + # exit on ctrl-c + except KeyboardInterrupt: + sys.exit() + +if __name__ == '__main__': + # Skip the first line which contains the version header. + print_line(read_line()) + + # The second line contains the start of the infinite array. + print_line(read_line()) + + while True: + line, prefix = read_line(), '' + # ignore comma at start of lines + if line.startswith(','): + line, prefix = line[1:], ',' + + j = json.loads(line) + # insert information into the start of the json, but could be anywhere + # CHANGE THIS LINE TO INSERT SOMETHING ELSE + j.insert(0, {'full_text' : '%s' % get_governor(), 'name' : 'gov'}) + # and echo back new encoded json + print_line(prefix+json.dumps(j)) -- 2.39.2