From 9e72fc48c81bb31017912bb0376530cbfa89d989 Mon Sep 17 00:00:00 2001
From: Levi Baber <baber@iastate.edu>
Date: Thu, 2 May 2019 11:04:21 -0500
Subject: [PATCH] fix error when adding package to env in container (#11321)

---
 lib/spack/spack/environment.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/spack/spack/environment.py b/lib/spack/spack/environment.py
index 62cad8545cc..02b0a6d93c5 100644
--- a/lib/spack/spack/environment.py
+++ b/lib/spack/spack/environment.py
@@ -125,7 +125,7 @@ def activate(
             cmds += 'setenv SPACK_OLD_PROMPT "${prompt}";\n'
             cmds += 'set prompt="%s ${prompt}";\n' % prompt
     else:
-        if 'color' in os.environ['TERM'] and prompt:
+        if os.getenv('TERM') and 'color' in os.getenv('TERM') and prompt:
             prompt = colorize('@G{%s} ' % prompt, color=True)
 
         cmds += 'export SPACK_ENV=%s;\n' % env.path