| from collections import defaultdict | |
| from logging import getLogger | |
| from typing import Any, DefaultDict | |
| from pip._vendor.resolvelib.reporters import BaseReporter | |
| from .base import Candidate, Requirement | |
| logger = getLogger(__name__) | |
| class PipReporter(BaseReporter): | |
| def __init__(self) -> None: | |
| self.backtracks_by_package: DefaultDict[str, int] = defaultdict(int) | |
| self._messages_at_backtrack = { | |
| 1: ( | |
| "pip is looking at multiple versions of {package_name} to " | |
| "determine which version is compatible with other " | |
| "requirements. This could take a while." | |
| ), | |
| 8: ( | |
| "pip is looking at multiple versions of {package_name} to " | |
| "determine which version is compatible with other " | |
| "requirements. This could take a while." | |
| ), | |
| 13: ( | |
| "This is taking longer than usual. You might need to provide " | |
| "the dependency resolver with stricter constraints to reduce " | |
| "runtime. See https://pip.pypa.io/warnings/backtracking for " | |
| "guidance. If you want to abort this run, press Ctrl + C." | |
| ), | |
| } | |
| def backtracking(self, candidate: Candidate) -> None: | |
| self.backtracks_by_package[candidate.name] += 1 | |
| count = self.backtracks_by_package[candidate.name] | |
| if count not in self._messages_at_backtrack: | |
| return | |
| message = self._messages_at_backtrack[count] | |
| logger.info("INFO: %s", message.format(package_name=candidate.name)) | |
| class PipDebuggingReporter(BaseReporter): | |
| """A reporter that does an info log for every event it sees.""" | |
| def starting(self) -> None: | |
| logger.info("Reporter.starting()") | |
| def starting_round(self, index: int) -> None: | |
| logger.info("Reporter.starting_round(%r)", index) | |
| def ending_round(self, index: int, state: Any) -> None: | |
| logger.info("Reporter.ending_round(%r, state)", index) | |
| def ending(self, state: Any) -> None: | |
| logger.info("Reporter.ending(%r)", state) | |
| def adding_requirement(self, requirement: Requirement, parent: Candidate) -> None: | |
| logger.info("Reporter.adding_requirement(%r, %r)", requirement, parent) | |
| def backtracking(self, candidate: Candidate) -> None: | |
| logger.info("Reporter.backtracking(%r)", candidate) | |
| def pinning(self, candidate: Candidate) -> None: | |
| logger.info("Reporter.pinning(%r)", candidate) | |