From 7277b973f03c6a4cffbbc57107d08e13d833695a Mon Sep 17 00:00:00 2001 From: Savio Fernandes Date: Fri, 16 Nov 2018 10:07:55 -0500 Subject: [PATCH] add option to skip undefined variables --- powerline_shell/segments/env.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/powerline_shell/segments/env.py b/powerline_shell/segments/env.py index 29ef048..5403b03 100644 --- a/powerline_shell/segments/env.py +++ b/powerline_shell/segments/env.py @@ -4,7 +4,11 @@ from ..utils import BasicSegment class Segment(BasicSegment): def add_to_powerline(self): - self.powerline.append( - " %s " % os.getenv(self.segment_def["var"]), + var_value = os.getenv(self.segment_def["var"]) + if (not var_value and + self.segment_def.get("skip_undefined", False)): + return + self.powerline.append( + " %s " % var_value, self.segment_def.get("fg_color", self.powerline.theme.PATH_FG), - self.segment_def.get("bg_color", self.powerline.theme.PATH_BG)) + self.segment_def.get("bg_color", self.powerline.theme.PATH_BG)) \ No newline at end of file