mirror of
https://github.com/qmk/qmk_firmware.git
synced 2025-07-14 11:51:32 +00:00
28 lines
739 B
Python
28 lines
739 B
Python
"""Utility functions.
|
|
"""
|
|
import contextlib
|
|
import functools
|
|
import multiprocessing
|
|
|
|
@contextlib.contextmanager
|
|
def parallelize(do_parallel):
|
|
"""Returns a function that can be used in place of a map() call.
|
|
|
|
Attempts to use `mpire`, falling back to `multiprocessing` if it's not
|
|
available. If parallelization is not requested, returns the original map()
|
|
function.
|
|
"""
|
|
if not do_parallel:
|
|
yield map
|
|
return
|
|
|
|
with contextlib.suppress(ImportError):
|
|
from mpire import WorkerPool
|
|
with WorkerPool() as pool:
|
|
yield functools.partial(pool.imap_unordered, progress_bar=True)
|
|
return
|
|
|
|
with multiprocessing.Pool() as pool:
|
|
yield pool.imap_unordered
|
|
return
|