153 lines
6 KiB
Python
153 lines
6 KiB
Python
import os
|
|
try:
|
|
import ast
|
|
except ImportError:
|
|
log.warning('--list modules will not work (module "ast" not found)')
|
|
import sys
|
|
import glob
|
|
import textwrap
|
|
import argparse
|
|
import logging
|
|
|
|
import core.theme
|
|
|
|
import util.store
|
|
import util.format
|
|
|
|
import modules.core
|
|
import modules.contrib
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
MODULE_HELP = 'Specify a space-separated list of modules to load. The order of the list determines their order in the i3bar (from left to right). Use <module>:<alias> to provide an alias in case you want to load the same module multiple times, but specify different parameters.'
|
|
PARAMETER_HELP = 'Provide configuration parameters in the form of <module>.<key>=<value>'
|
|
THEME_HELP = 'Specify the theme to use for drawing modules'
|
|
|
|
def all_modules():
|
|
"""Return a list of available modules"""
|
|
result = {}
|
|
|
|
for path in [ modules.core.__file__, modules.contrib.__file__ ]:
|
|
path = os.path.dirname(path)
|
|
for mod in glob.iglob('{}/*.py'.format(path)):
|
|
result[os.path.basename(mod).replace('.py', '')] = 1
|
|
|
|
res = list(result.keys())
|
|
res.sort()
|
|
return res
|
|
|
|
class print_usage(argparse.Action):
|
|
def __init__(self, option_strings, dest, nargs=None, **kwargs):
|
|
argparse.Action.__init__(self, option_strings, dest, nargs, **kwargs)
|
|
self._indent = ' '*2
|
|
|
|
def __call__(self, parser, namespace, value, option_string=None):
|
|
if value == 'modules':
|
|
self._args = namespace
|
|
self._format = 'plain'
|
|
self.print_modules()
|
|
elif value == 'modules-markdown':
|
|
self._args = namespace
|
|
self._format = 'markdown'
|
|
self.print_modules()
|
|
elif value == 'themes':
|
|
self.print_themes()
|
|
sys.exit(0)
|
|
|
|
def print_themes(self):
|
|
print(', '.join(core.theme.themes()))
|
|
|
|
def print_modules(self):
|
|
if self._format == 'markdown':
|
|
print('# Table of modules')
|
|
print('|Name |Description |')
|
|
print('|-----|------------|')
|
|
|
|
for m in all_modules():
|
|
try:
|
|
basepath = os.path.abspath(os.path.join(os.path.dirname(os.path.realpath(__file__)), '..'))
|
|
filename = os.path.join(basepath, 'modules', 'core', '{}.py'.format(m))
|
|
if not os.path.exists(filename):
|
|
filename = os.path.join(basepath, 'modules', 'contrib', '{}.py'.format(m))
|
|
if not os.path.exists(filename):
|
|
log.warning('module {} not found'.format(m))
|
|
continue
|
|
|
|
doc = None
|
|
with open(filename) as f:
|
|
tree = ast.parse(f.read())
|
|
doc = ast.get_docstring(tree)
|
|
|
|
if not doc:
|
|
log.warning('failed to find docstring for {}'.format(m))
|
|
continue
|
|
if self._format == 'markdown':
|
|
doc = doc.replace('<', '\<')
|
|
doc = doc.replace('>', '\>')
|
|
doc = doc.replace('\n', '<br>')
|
|
print('|{} |{} |'.format(m, doc))
|
|
else:
|
|
print(textwrap.fill('{}:'.format(m), 80,
|
|
initial_indent=self._indent*2, subsequent_indent=self._indent*2))
|
|
for line in doc.split('\n'):
|
|
print(textwrap.fill(line, 80,
|
|
initial_indent=self._indent*3, subsequent_indent=self._indent*6))
|
|
except Exception as e:
|
|
log.warning(e)
|
|
|
|
class Config(util.store.Store):
|
|
def __init__(self, args):
|
|
super(Config, self).__init__()
|
|
|
|
parser = argparse.ArgumentParser(description='bumblebee-status is a modular, theme-able status line generator for the i3 window manager. https://github.com/tobi-wan-kenobi/bumblebee-status/wiki')
|
|
parser.add_argument('-m', '--modules', nargs='+', action='append', default=[],
|
|
help=MODULE_HELP)
|
|
parser.add_argument('-p', '--parameters', nargs='+', action='append', default=[],
|
|
help=PARAMETER_HELP)
|
|
parser.add_argument('-t', '--theme', default='default', help=THEME_HELP)
|
|
parser.add_argument('-i', '--iconset', default='auto',
|
|
help='Specify the name of an iconset to use (overrides theme default)')
|
|
parser.add_argument('-a', '--autohide', nargs='+', default=[],
|
|
help='Specify a list of modules to hide when not in warning/error state')
|
|
parser.add_argument('-d', '--debug', action='store_true',
|
|
help='Add debug fields to i3 output')
|
|
parser.add_argument('-f', '--logfile', help='destination for the debug log file, if -d|--debug is specified; defaults to stderr')
|
|
parser.add_argument('-r', '--right-to-left', action='store_true', help='Draw widgets from right to left, rather than left to right (which is the default)')
|
|
parser.add_argument('-l', '--list', choices=['modules', 'themes', 'modules-markdown'], help='Display a list of available themes or available modules, along with their parameters',
|
|
action=print_usage)
|
|
|
|
self.__args = parser.parse_args(args)
|
|
|
|
parameters = [ item for sub in self.__args.parameters for item in sub ]
|
|
for param in parameters:
|
|
if not '=' in param:
|
|
log.error('missing value for parameter "{}" - ignoring this parameter'.format(param))
|
|
continue
|
|
key, value = param.split('=', 1)
|
|
self.set(key, value)
|
|
|
|
def modules(self):
|
|
return [item for sub in self.__args.modules for item in sub]
|
|
|
|
def interval(self, default=1):
|
|
return util.format.seconds(self.get('interval', default))
|
|
|
|
def debug(self):
|
|
return self.__args.debug
|
|
|
|
def reverse(self):
|
|
return self.__args.right_to_left
|
|
|
|
def logfile(self):
|
|
return self.__args.logfile
|
|
|
|
def theme(self):
|
|
return self.__args.theme
|
|
|
|
def iconset(self):
|
|
return self.__args.iconset
|
|
|
|
def autohide(self, name):
|
|
return name in self.__args.autohide
|
|
|
|
# vim: tabstop=8 expandtab shiftwidth=4 softtabstop=4
|