Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/pip/_internal/req/req_set.py: 46%

1import logging 

2from collections import OrderedDict 

3from typing import Dict, List 

4 

5from pip._vendor.packaging.utils import canonicalize_name 

6 

7from pip._internal.req.req_install import InstallRequirement 

8 

9logger = logging.getLogger(__name__) 

10 

11 

12class RequirementSet: 

13 def __init__(self, check_supported_wheels: bool = True) -> None: 

14 """Create a RequirementSet.""" 

15 

16 self.requirements: Dict[str, InstallRequirement] = OrderedDict() 

17 self.check_supported_wheels = check_supported_wheels 

18 

19 self.unnamed_requirements: List[InstallRequirement] = [] 

20 

21 def __str__(self) -> str: 

22 requirements = sorted( 

23 (req for req in self.requirements.values() if not req.comes_from), 

24 key=lambda req: canonicalize_name(req.name or ""), 

25 ) 

26 return " ".join(str(req.req) for req in requirements) 

27 

28 def __repr__(self) -> str: 

29 requirements = sorted( 

30 self.requirements.values(), 

31 key=lambda req: canonicalize_name(req.name or ""), 

32 ) 

33 

34 format_string = "<{classname} object; {count} requirement(s): {reqs}>" 

35 return format_string.format( 

36 classname=self.__class__.__name__, 

37 count=len(requirements), 

38 reqs=", ".join(str(req.req) for req in requirements), 

39 ) 

40 

41 def add_unnamed_requirement(self, install_req: InstallRequirement) -> None: 

42 assert not install_req.name 

43 self.unnamed_requirements.append(install_req) 

44 

45 def add_named_requirement(self, install_req: InstallRequirement) -> None: 

46 assert install_req.name 

47 

48 project_name = canonicalize_name(install_req.name) 

49 self.requirements[project_name] = install_req 

50 

51 def has_requirement(self, name: str) -> bool: 

52 project_name = canonicalize_name(name) 

53 

54 return ( 

55 project_name in self.requirements 

56 and not self.requirements[project_name].constraint 

57 ) 

58 

59 def get_requirement(self, name: str) -> InstallRequirement: 

60 project_name = canonicalize_name(name) 

61 

62 if project_name in self.requirements: 

63 return self.requirements[project_name] 

64 

65 raise KeyError(f"No project with the name {name!r}") 

66 

67 @property 

68 def all_requirements(self) -> List[InstallRequirement]: 

69 return self.unnamed_requirements + list(self.requirements.values()) 

70 

71 @property 

72 def requirements_to_install(self) -> List[InstallRequirement]: 

73 """Return the list of requirements that need to be installed. 

74 

75 TODO remove this property together with the legacy resolver, since the new 

76 resolver only returns requirements that need to be installed. 

77 """ 

78 return [ 

79 install_req 

80 for install_req in self.all_requirements 

81 if not install_req.constraint and not install_req.satisfied_by 

82 ]