if user is not in environ set to working directory.

This commit is contained in:
Raelon Masters
2020-05-10 19:30:08 -04:00
parent 5eb6a99509
commit 1e2b7499f8

View File

@@ -11,8 +11,10 @@ class TerminalDisplay:
self.term = True
self.w, self.y = os.get_terminal_size()[0], os.get_terminal_size()[1]
self.home = os.environ["HOME"]
breakpoint()
self.user = os.environ["USER"]
try:
self.user = os.environ["USER"]
except KeyError:
self.user = os.environ["PWD"]
self.version = "0.4.0"
self.slogan = "The Installer Initiative"
self.green = "\033[1;32m"