tyler.durden/powerline
1
0
Fork 0
mirror of https://github.com/powerline/powerline.git synced 2025-07-25 23:05:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Handle Py2/3 Unicode differences when writing to stdout

Browse Source 
Closes #94.
This commit is contained in:
Kim Silkebækken 2013-01-22 08:04:26 +01:00
parent 8355e3d670
commit 3ee6e55025
2 changed files with 16 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/powerline-prompt
View File

@ -2,12 +2,12 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
'''Powerline prompt script.''' '''Powerline prompt script.'''
import argparse import argparse
import sys
try: try:
from powerline.core import Powerline from powerline.core import Powerline
except ImportError: except ImportError:
import os import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from powerline.core import Powerline # NOQA from powerline.core import Powerline # NOQA
@ -18,8 +18,12 @@ parser.add_argument('--ext', default='shell')
if __name__ == '__main__': if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
pl = Powerline(ext=args.ext, renderer_module=args.renderer_module) powerline = Powerline(ext=args.ext, renderer_module=args.renderer_module)
segments = pl.renderer.get_theme().get_segments() segments = powerline.renderer.get_theme().get_segments()
if args.side != 'all': if args.side != 'all':
segments = [s for s in segments if s['side'] == args.side] segments = [s for s in segments if s['side'] == args.side]
print(pl.renderer.render(None, segments=segments)) rendered = powerline.renderer.render(None, segments=segments)
try:
sys.stdout.write(rendered)
except UnicodeEncodeError:
sys.stdout.write(rendered.encode('utf-8'))

12
scripts/powerline-tmux
View File

@ -2,12 +2,12 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
'''Powerline tmux statusline.''' '''Powerline tmux statusline.'''
import argparse import argparse
import sys
try: try:
from powerline.core import Powerline from powerline.core import Powerline
except ImportError: except ImportError:
import os import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from powerline.core import Powerline # NOQA from powerline.core import Powerline # NOQA
@ -17,8 +17,12 @@ parser.add_argument('--ext', default='tmux')
if __name__ == '__main__': if __name__ == '__main__':
args = parser.parse_args() args = parser.parse_args()
pl = Powerline(args.ext) powerline = Powerline(args.ext)
segments = pl.renderer.get_theme().get_segments() segments = powerline.renderer.get_theme().get_segments()
if args.side != 'all': if args.side != 'all':
segments = [s for s in segments if s['side'] == args.side] segments = [s for s in segments if s['side'] == args.side]
print(pl.renderer.render(None, segments=segments)) rendered = powerline.renderer.render(None, segments=segments)
try:
sys.stdout.write(rendered)
except UnicodeEncodeError:
sys.stdout.write(rendered.encode('utf-8'))