Allow spawn() and friends to be used as context managers

Closes gh-111

1 files changed, 9 insertions, 0 deletions

diff --git a/pexpect/spawnbase.py b/pexpect/spawnbase.py
index b1bc2f3..d79c5c0 100644
--- a/pexpect/spawnbase.py
+++ b/pexpect/spawnbase.py
@@ -443,6 +443,15 @@ class SpawnBase(object):
         """Overridden in subclass using tty"""
         return False
 
+    # For 'with spawn(...) as child:'
+    def __enter__(self):
+        return self
+    
+    def __exit__(self, etype, evalue, tb):
+        # We rely on subclasses to implement close(). If they don't, it's not
+        # clear what a context manager should do.
+        self.close()
+
 class SpawnBaseUnicode(SpawnBase):
     if PY3:
         string_type = str