?HAMA

Navigation

Breadcrumbs

Create

Edit File Size: 3.12 KB

//opt/alt/python310/lib/python3.10/site-packages/pip/_internal/utils/parallel.py

Text Base64

"""Convenient parallelization of higher order functions.

This module provides two helper functions, with appropriate fallbacks on
Python 2 and on systems lacking support for synchronization mechanisms:

- map_multiprocess
- map_multithread

These helpers work like Python 3's map, with two differences:

- They don't guarantee the order of processing of
  the elements of the iterable.
- The underlying process/thread pools chop the iterable into
  a number of chunks, so that for very long iterables using
  a large value for chunksize can make the job complete much faster
  than using the default value of 1.
"""

__all__ = ["map_multiprocess", "map_multithread"]

from contextlib import contextmanager
from multiprocessing import Pool as ProcessPool
from multiprocessing import pool
from multiprocessing.dummy import Pool as ThreadPool
from typing import Callable, Iterable, Iterator, TypeVar, Union

from pip._vendor.requests.adapters import DEFAULT_POOLSIZE

Pool = Union[pool.Pool, pool.ThreadPool]
S = TypeVar("S")
T = TypeVar("T")

# On platforms without sem_open, multiprocessing[.dummy] Pool
# cannot be created.
try:
    import multiprocessing.synchronize  # noqa
except ImportError:
    LACK_SEM_OPEN = True
else:
    LACK_SEM_OPEN = False

# Incredibly large timeout to work around bpo-8296 on Python 2.
TIMEOUT = 2000000

@contextmanager
def closing(pool: Pool) -> Iterator[Pool]:
    """Return a context manager making sure the pool closes properly."""
    try:
        yield pool
    finally:
        # For Pool.imap*, close and join are needed
        # for the returned iterator to begin yielding.
        pool.close()
        pool.join()
        pool.terminate()

def _map_fallback(
    func: Callable[[S], T], iterable: Iterable[S], chunksize: int = 1
) -> Iterator[T]:
    """Make an iterator applying func to each element in iterable.

This function is the sequential fallback either on Python 2
    where Pool.imap* doesn't react to KeyboardInterrupt
    or when sem_open is unavailable.
    """
    return map(func, iterable)

def _map_multiprocess(
    func: Callable[[S], T], iterable: Iterable[S], chunksize: int = 1
) -> Iterator[T]:
    """Chop iterable into chunks and submit them to a process pool.

For very long iterables using a large value for chunksize can make
    the job complete much faster than using the default value of 1.

Return an unordered iterator of the results.
    """
    with closing(ProcessPool()) as pool:
        return pool.imap_unordered(func, iterable, chunksize)

def _map_multithread(
    func: Callable[[S], T], iterable: Iterable[S], chunksize: int = 1
) -> Iterator[T]:
    """Chop iterable into chunks and submit them to a thread pool.

For very long iterables using a large value for chunksize can make
    the job complete much faster than using the default value of 1.

Return an unordered iterator of the results.
    """
    with closing(ThreadPool(DEFAULT_POOLSIZE)) as pool:
        return pool.imap_unordered(func, iterable, chunksize)

if LACK_SEM_OPEN:
    map_multiprocess = map_multithread = _map_fallback
else:
    map_multiprocess = _map_multiprocess
    map_multithread = _map_multithread

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2026-02-10 08:08:40	Edit Download recursive recursive
appdirs.py	1.63 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
compat.py	1.84 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
compatibility_tags.py	5.25 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
datetime.py	242 B	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
deprecation.py	3.54 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
direct_url_helpers.py	3.13 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
distutils_args.py	1.22 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
egg_link.py	2.15 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
encoding.py	1.14 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
entrypoints.py	1.10 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
filesystem.py	5.75 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
filetypes.py	716 B	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
glibc.py	3.04 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
hashes.py	4.70 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
inject_securetransport.py	795 B	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
logging.py	11.26 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
misc.py	20.29 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
models.py	1.17 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
packaging.py	2.88 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
parallel.py	3.12 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
pkg_resources.py	987 B	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
setuptools_build.py	4.59 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
subprocess.py	9.82 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
temp_dir.py	7.48 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
unpacking.py	8.70 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
urls.py	1.72 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
virtualenv.py	3.38 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
wheel.py	6.02 KB	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
_log.py	1015 B	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive
__init__.py	0 B	lrw-r--r--	2026-01-09 15:17:41	Edit Download recursive recursive

Directory Contents