commit 44c66c72c9045a7824d07175cfc07845e4d7d28a
author Serhiy Storchaka <storchaka@gmail.com> Sat Dec 29 22:54:49 2012 +0200
committer Serhiy Storchaka <storchaka@gmail.com> Sat Dec 29 22:54:49 2012 +0200
tree a88a17bd7c1a4fb9542d5d9a4f7d91c80eb303e4
parent ff2840a76f04c8846e8ad5b7e8379998ab3e87a6

Issue #16486: Make aifc files work with 'with' as context managers.

Lib/aifc.py

diff --git a/Lib/aifc.py b/Lib/aifc.py
index a19b38f..67ea5da 100644
--- a/Lib/aifc.py
+++ b/Lib/aifc.py
@@ -334,6 +334,12 @@
         # else, assume it is an open file object already
         self.initfp(f)
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *args):
+        self.close()
+
     #
     # User visible methods.
     #
@@ -553,6 +559,12 @@
     def __del__(self):
         self.close()
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *args):
+        self.close()
+
     #
     # User visible methods.
     #