add option to skip undefined variables

This commit is contained in:
Savio Fernandes 2018-11-16 10:07:55 -05:00
parent a9b8c9bb39
commit 7277b973f0

View file

@ -4,7 +4,11 @@ from ..utils import BasicSegment
class Segment(BasicSegment):
def add_to_powerline(self):
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 " % os.getenv(self.segment_def["var"]),
" %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))