Guard against USER not existing in the environment

rust-lang · Feb 2, 2017 · 8e793eb · 8e793eb
1 parent a0efdf3
commit 8e793eb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/bootstrap/bootstrap.py b/src/bootstrap/bootstrap.py
@@ -439,7 +439,7 @@ def main():
                               'CFG_ENABLE_VENDOR' in rb.config_mk
 
     if 'SUDO_USER' in os.environ and not rb.use_vendored_sources:
-        if os.environ['USER'] != os.environ['SUDO_USER']:
+        if os.environ.get('USER') != os.environ['SUDO_USER']:
             rb.use_vendored_sources = True
             print('info: looks like you are running this command under `sudo`')
             print('      and so in order to preserve your $HOME this will now')