zsh/b-ryan.powerline-shell
1
0
Fork 0
mirror of https://github.com/b-ryan/powerline-shell.git synced 2026-09-10 07:26:28 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix package missing in cygwin

Browse source 
resolves #323
This commit is contained in:
ericLemanissier 2017-12-11 12:44:47 +01:00 committed by GitHub
parent b6be8abb19
commit fba09ed096
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
powerline_shell/segments/username.py
View file

@ -1,6 +1,10 @@
from ..utils import BasicSegment
import os
import pwd
try:
import pwd
except ImportError:
pwd = None
import getpass
class Segment(BasicSegment):
@ -13,7 +17,12 @@ class Segment(BasicSegment):
else:
user_prompt = " %s " % os.getenv("USER")
if pwd.getpwuid(os.getuid())[0] == "root":
if pwd:
user_name = pwd.getpwuid(os.getuid())[0]
else:
user_name = getpass.getuser()
if user_name == "root":
bgcolor = powerline.theme.USERNAME_ROOT_BG
else:
bgcolor = powerline.theme.USERNAME_BG