From 2b0c01482b8bcd509e0cb18b07c33bcee8dc5fc8 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kim=20Silkeb=C3=A6kken?= Date: Tue, 22 Jan 2013 09:04:57 +0100 Subject: [PATCH] Improve side handling in prompt/tmux scripts --- scripts/powerline-prompt | 7 ++----- scripts/powerline-tmux | 7 ++----- 2 files changed, 4 insertions(+), 10 deletions(-) diff --git a/scripts/powerline-prompt b/scripts/powerline-prompt index 50efcfb6..1670d882 100755 --- a/scripts/powerline-prompt +++ b/scripts/powerline-prompt @@ -12,17 +12,14 @@ except ImportError: from powerline.core import Powerline # NOQA parser = argparse.ArgumentParser(description=__doc__) -parser.add_argument('side', nargs='?', default='all', choices=('all', 'left', 'right')) +parser.add_argument('side', nargs='?', default=None, choices=('left', 'right')) parser.add_argument('--renderer_module', default='shell', choices=('shell', 'zsh_prompt')) parser.add_argument('--ext', default='shell') if __name__ == '__main__': args = parser.parse_args() powerline = Powerline(ext=args.ext, renderer_module=args.renderer_module) - segments = powerline.renderer.get_theme().get_segments() - if args.side != 'all': - segments = [s for s in segments if s['side'] == args.side] - rendered = powerline.renderer.render(None, segments=segments) + rendered = powerline.renderer.render(None, side=args.side) try: sys.stdout.write(rendered) except UnicodeEncodeError: diff --git a/scripts/powerline-tmux b/scripts/powerline-tmux index 0b9e1377..51e01755 100755 --- a/scripts/powerline-tmux +++ b/scripts/powerline-tmux @@ -12,16 +12,13 @@ except ImportError: from powerline.core import Powerline # NOQA parser = argparse.ArgumentParser(description=__doc__) -parser.add_argument('side', nargs='?', default='all', choices=('all', 'left', 'right')) +parser.add_argument('side', nargs='?', default=None, choices=('left', 'right')) parser.add_argument('--ext', default='tmux') if __name__ == '__main__': args = parser.parse_args() powerline = Powerline(args.ext) - segments = powerline.renderer.get_theme().get_segments() - if args.side != 'all': - segments = [s for s in segments if s['side'] == args.side] - rendered = powerline.renderer.render(None, segments=segments) + rendered = powerline.renderer.render(None, side=args.side) try: sys.stdout.write(rendered) except UnicodeEncodeError: