mirror of
https://github.com/powerline/powerline.git
synced 2025-07-16 18:34:57 +02:00
76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
# vim:fileencoding=utf-8:noet
|
|
|
|
from powerline import Powerline
|
|
from powerline.lib import mergedicts, parsedotval
|
|
|
|
|
|
def mergeargs(argvalue):
|
|
if not argvalue:
|
|
return None
|
|
r = {}
|
|
for subval in argvalue:
|
|
mergedicts(r, dict([subval]))
|
|
return r
|
|
|
|
|
|
class ShellPowerline(Powerline):
|
|
def __init__(self, args, **kwargs):
|
|
self.args = args
|
|
self.theme_option = args.theme_option
|
|
super(ShellPowerline, self).__init__(args.ext[0], args.renderer_module, **kwargs)
|
|
|
|
def load_main_config(self):
|
|
r = super(ShellPowerline, self).load_main_config()
|
|
if self.args.config:
|
|
mergedicts(r, self.args.config)
|
|
return r
|
|
|
|
def load_theme_config(self, name):
|
|
r = super(ShellPowerline, self).load_theme_config(name)
|
|
if self.theme_option and name in self.theme_option:
|
|
mergedicts(r, self.theme_option[name])
|
|
return r
|
|
|
|
def get_config_paths(self):
|
|
if self.args.config_path:
|
|
return [self.args.config_path]
|
|
else:
|
|
return super(ShellPowerline, self).get_config_paths()
|
|
|
|
def get_local_themes(self, local_themes):
|
|
if not local_themes:
|
|
return {}
|
|
|
|
return dict(((key, {'config': self.load_theme_config(val)})
|
|
for key, val in local_themes.items()))
|
|
|
|
|
|
def get_argparser(parser=None, *args, **kwargs):
|
|
if not parser:
|
|
import argparse
|
|
parser = argparse.ArgumentParser
|
|
p = parser(*args, **kwargs)
|
|
p.add_argument('ext', nargs=1)
|
|
p.add_argument('side', nargs='?', choices=('left', 'right'))
|
|
p.add_argument('-r', '--renderer_module', metavar='MODULE', type=str)
|
|
p.add_argument('-w', '--width', type=int)
|
|
p.add_argument('--last_exit_code', metavar='INT', type=int)
|
|
p.add_argument('--last_pipe_status', metavar='LIST', default='', type=lambda s: [int(status) for status in s.split()])
|
|
p.add_argument('--jobnum', metavar='INT', type=int)
|
|
p.add_argument('-c', '--config', metavar='KEY.KEY=VALUE', action='append')
|
|
p.add_argument('-t', '--theme_option', metavar='THEME.KEY.KEY=VALUE', action='append')
|
|
p.add_argument('-p', '--config_path', metavar='PATH')
|
|
p.add_argument('-R', '--renderer_arg', metavar='KEY=VAL', action='append')
|
|
return p
|
|
|
|
|
|
def finish_args(args):
|
|
if args.config:
|
|
args.config = mergeargs((parsedotval(v) for v in args.config))
|
|
if args.theme_option:
|
|
args.theme_option = mergeargs((parsedotval(v) for v in args.theme_option))
|
|
else:
|
|
args.theme_option = {}
|
|
if args.renderer_arg:
|
|
args.renderer_arg = mergeargs((parsedotval(v) for v in args.renderer_arg))
|