You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.0 KiB
105 lines
3.0 KiB
from __future__ import annotations
|
|
|
|
import collections
|
|
import logging
|
|
from collections.abc import Generator, Sequence
|
|
from dataclasses import dataclass
|
|
|
|
from pip._internal.cli.progress_bars import BarType, get_install_progress_renderer
|
|
from pip._internal.utils.logging import indent_log
|
|
|
|
from .req_file import parse_requirements
|
|
from .req_install import InstallRequirement
|
|
from .req_set import RequirementSet
|
|
|
|
__all__ = [
|
|
"RequirementSet",
|
|
"InstallRequirement",
|
|
"parse_requirements",
|
|
"install_given_reqs",
|
|
]
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class InstallationResult:
|
|
name: str
|
|
|
|
|
|
def _validate_requirements(
|
|
requirements: list[InstallRequirement],
|
|
) -> Generator[tuple[str, InstallRequirement], None, None]:
|
|
for req in requirements:
|
|
assert req.name, f"invalid to-be-installed requirement: {req}"
|
|
yield req.name, req
|
|
|
|
|
|
def install_given_reqs(
|
|
requirements: list[InstallRequirement],
|
|
global_options: Sequence[str],
|
|
root: str | None,
|
|
home: str | None,
|
|
prefix: str | None,
|
|
warn_script_location: bool,
|
|
use_user_site: bool,
|
|
pycompile: bool,
|
|
progress_bar: BarType,
|
|
) -> list[InstallationResult]:
|
|
"""
|
|
Install everything in the given list.
|
|
|
|
(to be called after having downloaded and unpacked the packages)
|
|
"""
|
|
to_install = collections.OrderedDict(_validate_requirements(requirements))
|
|
|
|
if to_install:
|
|
logger.info(
|
|
"Installing collected packages: %s",
|
|
", ".join(to_install.keys()),
|
|
)
|
|
|
|
installed = []
|
|
|
|
show_progress = logger.isEnabledFor(logging.INFO) and len(to_install) > 1
|
|
|
|
items = iter(to_install.values())
|
|
if show_progress:
|
|
renderer = get_install_progress_renderer(
|
|
bar_type=progress_bar, total=len(to_install)
|
|
)
|
|
items = renderer(items)
|
|
|
|
with indent_log():
|
|
for requirement in items:
|
|
req_name = requirement.name
|
|
assert req_name is not None
|
|
if requirement.should_reinstall:
|
|
logger.info("Attempting uninstall: %s", req_name)
|
|
with indent_log():
|
|
uninstalled_pathset = requirement.uninstall(auto_confirm=True)
|
|
else:
|
|
uninstalled_pathset = None
|
|
|
|
try:
|
|
requirement.install(
|
|
global_options,
|
|
root=root,
|
|
home=home,
|
|
prefix=prefix,
|
|
warn_script_location=warn_script_location,
|
|
use_user_site=use_user_site,
|
|
pycompile=pycompile,
|
|
)
|
|
except Exception:
|
|
# if install did not succeed, rollback previous uninstall
|
|
if uninstalled_pathset and not requirement.install_succeeded:
|
|
uninstalled_pathset.rollback()
|
|
raise
|
|
else:
|
|
if uninstalled_pathset and requirement.install_succeeded:
|
|
uninstalled_pathset.commit()
|
|
|
|
installed.append(InstallationResult(req_name))
|
|
|
|
return installed
|
|
|