summaryrefslogtreecommitdiffstats
path: root/Lib/multiprocessing/pool.py
diff options
context:
space:
mode:
authorRichard Oudkerk <shibturn@gmail.com>2012-06-18 16:47:52 (GMT)
committerRichard Oudkerk <shibturn@gmail.com>2012-06-18 16:47:52 (GMT)
commitd69cfe88eae5b177b1aaf51c39e85fb92c34cf22 (patch)
tree43151241e2aab605f65e3634affd630ea6e084a3 /Lib/multiprocessing/pool.py
parent0f884273b0319272ca32d529af5c761d6dfa8a34 (diff)
downloadcpython-d69cfe88eae5b177b1aaf51c39e85fb92c34cf22.zip
cpython-d69cfe88eae5b177b1aaf51c39e85fb92c34cf22.tar.gz
cpython-d69cfe88eae5b177b1aaf51c39e85fb92c34cf22.tar.bz2
Issue #15064: Implement context manager protocol for multiprocessing types
Diffstat (limited to 'Lib/multiprocessing/pool.py')
-rw-r--r--Lib/multiprocessing/pool.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/Lib/multiprocessing/pool.py b/Lib/multiprocessing/pool.py
index 59e547a..9e07e32 100644
--- a/Lib/multiprocessing/pool.py
+++ b/Lib/multiprocessing/pool.py
@@ -522,6 +522,12 @@ class Pool(object):
debug('cleaning up worker %d' % p.pid)
p.join()
+ def __enter__(self):
+ return self
+
+ def __exit__(self, exc_type, exc_val, exc_tb):
+ self.terminate()
+
#
# Class whose instances are returned by `Pool.apply_async()`
#