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