| 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) |