From 542d221b5f39c55a1785bcbda415ab31bd6318a7 Mon Sep 17 00:00:00 2001
From: Antoine Pitrou <solipsis@pitrou.net>
Date: Wed, 9 Dec 2009 00:01:27 +0000
Subject: Issue #7461: objects returned by os.popen() should support the
 context manager protocol

---
 Lib/os.py | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'Lib/os.py')

diff --git a/Lib/os.py b/Lib/os.py
index ec5d280b9e..2a9937f12f 100644
--- a/Lib/os.py
+++ b/Lib/os.py
@@ -650,6 +650,10 @@ class _wrap_close:
             return returncode
         else:
             return returncode << 8  # Shift left to match old behavior
+    def __enter__(self):
+        return self
+    def __exit__(self, *args):
+        self.close()
     def __getattr__(self, name):
         return getattr(self._stream, name)
     def __iter__(self):
-- 
cgit v1.2.1