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