From daa66bc6acd2430cdd05e9883dd1eab69b698252 Mon Sep 17 00:00:00 2001
From: Andy Grover <agrover@redhat.com>
Date: Tue, 23 May 2017 12:00:10 -0700
Subject: [PATCH] Handle if TERM is not set

Previous patch causes problems otherwise in some cases.

Signed-off-by: Andy Grover <agrover@redhat.com>
---
 configshell/shell.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/configshell/shell.py b/configshell/shell.py
index c916366..e029c76 100644
--- a/configshell/shell.py
+++ b/configshell/shell.py
@@ -50,13 +50,14 @@ else:
     tty=False
     
     # remember the original setting
-    oldTerm = os.environ['TERM']
+    oldTerm = os.environ.get('TERM')
     os.environ['TERM'] = ''
 
     import readline
 
     # restore the orignal TERM setting
-    os.environ['TERM'] = oldTerm
+    if oldTerm != None:
+        os.environ['TERM'] = oldTerm
     del oldTerm
     
 # Pyparsing helper to group the location of a token and its value