use different color for system users (0 < EUID < 1000)

This commit is contained in:
Martinez 2016-02-01 20:25:45 +01:00
parent b273c0bd81
commit 934b9cd4cc

View file

@ -13,6 +13,7 @@ COLOR_LOAD_OK = 2
COLOR_LOAD_WARN = 3 COLOR_LOAD_WARN = 3
COLOR_LOAD_ERROR = 1 COLOR_LOAD_ERROR = 1
COLOR_USER_USER = 10 COLOR_USER_USER = 10
COLOR_USER_DAEMON = 32
COLOR_USER_ROOT = 196 COLOR_USER_ROOT = 196
COLOR_SESSION_LOCAL = 7 COLOR_SESSION_LOCAL = 7
COLOR_SESSION_REMOTE = 202 COLOR_SESSION_REMOTE = 202
@ -126,6 +127,7 @@ class Settings:
self.space_warn = float(argv[6]) self.space_warn = float(argv[6])
self.space_error = float(argv[7]) self.space_error = float(argv[7])
self.is_root = os.geteuid() == 0 self.is_root = os.geteuid() == 0
self.is_daemon = os.geteuid() < 1000
except: except:
print("Usage: %s $COLUMNS $PWD last_refresh_time load_warn load_error space_warn space_error") print("Usage: %s $COLUMNS $PWD last_refresh_time load_warn load_error space_warn space_error")
print(" e.g. %s 1 $COLUMNS $PWD 25 1.25 2.5 0.15 0.05") print(" e.g. %s 1 $COLUMNS $PWD 25 1.25 2.5 0.15 0.05")
@ -180,6 +182,8 @@ class LoginPart(Part):
# username # username
if settings.is_root: if settings.is_root:
self.fragments.append(style_color(COLOR_USER_ROOT)) self.fragments.append(style_color(COLOR_USER_ROOT))
elif settings.is_daemon:
self.fragments.append(style_color(COLOR_USER_DAEMON))
else: else:
self.fragments.append(style_color(COLOR_USER_USER)) self.fragments.append(style_color(COLOR_USER_USER))
@ -387,6 +391,7 @@ class GitPart(Part):
deep_parse = not any([cwd.startswith(d) for d in GIT_BLACKLIST]) deep_parse = not any([cwd.startswith(d) for d in GIT_BLACKLIST])
branch_name = command("git name-rev --name-only --no-undefined --always HEAD").strip() branch_name = command("git name-rev --name-only --no-undefined --always HEAD").strip()
#branch_name = command("git symbolic-ref HEAD").strip()#.remove("refs/heads/")
if branch_name: if branch_name:
self.fragments.append("") self.fragments.append("")