file formatting for username segment

This commit is contained in:
Buck Ryan 2017-09-05 21:47:03 -04:00
parent 0b02a4baad
commit d23f394350

View file

@ -1,18 +1,18 @@
from ..utils import BasicSegment
import os
class Segment(BasicSegment):
def add_to_powerline(self):
powerline = self.powerline
import os
if powerline.args.shell == 'bash':
user_prompt = ' \\u '
elif powerline.args.shell == 'zsh':
user_prompt = ' %n '
if powerline.args.shell == "bash":
user_prompt = r" \u "
elif powerline.args.shell == "zsh":
user_prompt = " %n "
else:
user_prompt = ' %s ' % os.getenv('USER')
user_prompt = " %s " % os.getenv("USER")
if os.getenv('USER') == 'root':
if os.getenv("USER") == "root":
bgcolor = powerline.theme.USERNAME_ROOT_BG
else:
bgcolor = powerline.theme.USERNAME_BG