def __enter__(self):
if self.verbose:
print "Creating process pool"
if self.use_multiprocessing:
self.process_pool = Pool()
else:
self.process_pool = FakePool()
return self.process_pool
class ProcessManager():
def __init__(self,use_multiprocessing,verbose=False):
self.use_multiprocessing = use_multiprocessing
self.verbose = verbose
def __enter__(self):
if self.verbose:
print "Creating process pool"
if self.use_multiprocessing:
self.process_pool = Pool()
else:
self.process_pool = FakePool()
return self.process_pool
def __exit__(self,errtype,val,traceback):
if self.verbose:
print "Closing and joining process pool"
self.process_pool.close()
self.process_pool.join()
return False
