git.reasoningtechnology.com Git - SubU/blob

   1 from collections import defaultdict
   2 from logging import getLogger
   3 from typing import Any, DefaultDict
   4
   5 from pip._vendor.resolvelib.reporters import BaseReporter
   6
   7 from .base import Candidate, Requirement
   8
   9 logger = getLogger(__name__)
  10
  11
  12 class PipReporter(BaseReporter):
  13     def __init__(self) -> None:
  14         self.backtracks_by_package: DefaultDict[str, int] = defaultdict(int)
  15
  16         self._messages_at_backtrack = {
  17             1: (
  18                 "pip is looking at multiple versions of {package_name} to "
  19                 "determine which version is compatible with other "
  20                 "requirements. This could take a while."
  21             ),
  22             8: (
  23                 "pip is looking at multiple versions of {package_name} to "
  24                 "determine which version is compatible with other "
  25                 "requirements. This could take a while."
  26             ),
  27             13: (
  28                 "This is taking longer than usual. You might need to provide "
  29                 "the dependency resolver with stricter constraints to reduce "
  30                 "runtime. See https://pip.pypa.io/warnings/backtracking for "
  31                 "guidance. If you want to abort this run, press Ctrl + C."
  32             ),
  33         }
  34
  35     def backtracking(self, candidate: Candidate) -> None:
  36         self.backtracks_by_package[candidate.name] += 1
  37
  38         count = self.backtracks_by_package[candidate.name]
  39         if count not in self._messages_at_backtrack:
  40             return
  41
  42         message = self._messages_at_backtrack[count]
  43         logger.info("INFO: %s", message.format(package_name=candidate.name))
  44
  45
  46 class PipDebuggingReporter(BaseReporter):
  47     """A reporter that does an info log for every event it sees."""
  48
  49     def starting(self) -> None:
  50         logger.info("Reporter.starting()")
  51
  52     def starting_round(self, index: int) -> None:
  53         logger.info("Reporter.starting_round(%r)", index)
  54
  55     def ending_round(self, index: int, state: Any) -> None:
  56         logger.info("Reporter.ending_round(%r, state)", index)
  57
  58     def ending(self, state: Any) -> None:
  59         logger.info("Reporter.ending(%r)", state)
  60
  61     def adding_requirement(self, requirement: Requirement, parent: Candidate) -> None:
  62         logger.info("Reporter.adding_requirement(%r, %r)", requirement, parent)
  63
  64     def backtracking(self, candidate: Candidate) -> None:
  65         logger.info("Reporter.backtracking(%r)", candidate)
  66
  67     def pinning(self, candidate: Candidate) -> None:
  68         logger.info("Reporter.pinning(%r)", candidate)