113 lines
3.9 KiB
Python
Executable file
113 lines
3.9 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
import glob
|
|
import pkgutil
|
|
import argparse
|
|
import textwrap
|
|
import importlib
|
|
import bumblebee.theme
|
|
import bumblebee.modules
|
|
import bumblebee.outputs.i3
|
|
|
|
def print_module_list():
|
|
print "available modules:"
|
|
path = os.path.dirname(bumblebee.modules.__file__)
|
|
for mod in [ name for _, name, _ in pkgutil.iter_modules([path])]:
|
|
m = importlib.import_module("bumblebee.modules.{}".format(mod))
|
|
|
|
desc = "n/a" if not hasattr(m, "description") else getattr(m, "description")()
|
|
usage = "n/a" if not hasattr(m, "usage") else getattr(m, "usage")()
|
|
notes = "n/a" if not hasattr(m, "notes") else getattr(m, "notes")()
|
|
|
|
print " {}: ".format(mod)
|
|
print textwrap.fill("Description: {}".format(desc),
|
|
80, initial_indent=" ", subsequent_indent=" ")
|
|
print textwrap.fill("Usage : {}".format(usage),
|
|
80, initial_indent=" ", subsequent_indent=" ")
|
|
print textwrap.fill("Notes : {}".format(notes),
|
|
80, initial_indent=" ", subsequent_indent=" ")
|
|
print ""
|
|
|
|
def print_theme_list():
|
|
d = bumblebee.theme.getpath()
|
|
|
|
print "available themes:"
|
|
print textwrap.fill(", ".join(
|
|
[ os.path.basename(f).replace(".json", "") for f in glob.iglob("{}/*.json".format(d)) ]),
|
|
80, initial_indent = " ", subsequent_indent = " "
|
|
)
|
|
|
|
def init_argument_parser():
|
|
parser = argparse.ArgumentParser(description="display system data in the i3bar")
|
|
parser.add_argument("-m", "--modules", nargs="+", help="List of modules to load. The order of the list determines their order in the i3bar (from left to right)", default=[])
|
|
parser.add_argument("-e", "--events", nargs="+", help="List of click events that should be handled. Format is: <module name><splitter, see -s><button ID><splitter><command to execute>", default=[])
|
|
parser.add_argument("-l", "--list", action="store_true", help="List all available modules and themes")
|
|
parser.add_argument("-t", "--theme", help="Specify which theme to use for drawing the modules")
|
|
parser.add_argument("-i", "--interval", help="Specify the update interval", default=1, type=int)
|
|
parser.add_argument("-s", "--split", help="Specify string to use for splitting modules and their arguments", default="::")
|
|
|
|
return parser
|
|
|
|
def getmodule(args, output, modulespec):
|
|
s = args.split
|
|
module_name = modulespec if not s in modulespec else modulespec.split(s)[0]
|
|
module_args = None if not s in modulespec else modulespec.split(s)[1:]
|
|
module = importlib.import_module("bumblebee.modules.{}".format(module_name))
|
|
return getattr(module, "Module")(output, module_args)
|
|
|
|
def register_event(args, output, event):
|
|
ev = event.split(args.split)
|
|
if len(ev) < 3:
|
|
sys.stderr.write("invalid format for click event, expect 3 parameters")
|
|
return
|
|
output.add_callback(
|
|
module=ev[0],
|
|
button=int(ev[1]),
|
|
cmd=ev[2],
|
|
)
|
|
|
|
def main():
|
|
parser = init_argument_parser()
|
|
|
|
if len(sys.argv) == 1:
|
|
parser.print_help()
|
|
parser.exit()
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.list:
|
|
print_module_list()
|
|
print_theme_list()
|
|
sys.exit(0)
|
|
|
|
theme = bumblebee.theme.Theme(args.theme) if args.theme else bumblebee.theme.Theme()
|
|
output = bumblebee.outputs.i3.i3bar(theme)
|
|
|
|
modules = []
|
|
for m in args.modules:
|
|
modules.append(getmodule(args, output, m))
|
|
|
|
for e in args.events:
|
|
register_event(args, output, e)
|
|
|
|
print output.start()
|
|
sys.stdout.flush()
|
|
|
|
while True:
|
|
theme.reset()
|
|
for m in modules:
|
|
output.add(m)
|
|
theme.next()
|
|
print output.get()
|
|
sys.stdout.flush()
|
|
time.sleep(args.interval)
|
|
|
|
print output.stop()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|