show.py 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181
  1. import logging
  2. import os
  3. from email.parser import FeedParser
  4. from optparse import Values
  5. from typing import Dict, Iterator, List
  6. from pip._vendor import pkg_resources
  7. from pip._vendor.packaging.utils import canonicalize_name
  8. from pip._internal.cli.base_command import Command
  9. from pip._internal.cli.status_codes import ERROR, SUCCESS
  10. from pip._internal.utils.misc import write_output
  11. logger = logging.getLogger(__name__)
  12. class ShowCommand(Command):
  13. """
  14. Show information about one or more installed packages.
  15. The output is in RFC-compliant mail header format.
  16. """
  17. usage = """
  18. %prog [options] <package> ..."""
  19. ignore_require_venv = True
  20. def add_options(self):
  21. # type: () -> None
  22. self.cmd_opts.add_option(
  23. '-f', '--files',
  24. dest='files',
  25. action='store_true',
  26. default=False,
  27. help='Show the full list of installed files for each package.')
  28. self.parser.insert_option_group(0, self.cmd_opts)
  29. def run(self, options, args):
  30. # type: (Values, List[str]) -> int
  31. if not args:
  32. logger.warning('ERROR: Please provide a package name or names.')
  33. return ERROR
  34. query = args
  35. results = search_packages_info(query)
  36. if not print_results(
  37. results, list_files=options.files, verbose=options.verbose):
  38. return ERROR
  39. return SUCCESS
  40. def search_packages_info(query):
  41. # type: (List[str]) -> Iterator[Dict[str, str]]
  42. """
  43. Gather details from installed distributions. Print distribution name,
  44. version, location, and installed files. Installed files requires a
  45. pip generated 'installed-files.txt' in the distributions '.egg-info'
  46. directory.
  47. """
  48. installed = {}
  49. for p in pkg_resources.working_set:
  50. installed[canonicalize_name(p.project_name)] = p
  51. query_names = [canonicalize_name(name) for name in query]
  52. missing = sorted(
  53. [name for name, pkg in zip(query, query_names) if pkg not in installed]
  54. )
  55. if missing:
  56. logger.warning('Package(s) not found: %s', ', '.join(missing))
  57. def get_requiring_packages(package_name):
  58. # type: (str) -> List[str]
  59. canonical_name = canonicalize_name(package_name)
  60. return [
  61. pkg.project_name for pkg in pkg_resources.working_set
  62. if canonical_name in
  63. [canonicalize_name(required.name) for required in
  64. pkg.requires()]
  65. ]
  66. for dist in [installed[pkg] for pkg in query_names if pkg in installed]:
  67. package = {
  68. 'name': dist.project_name,
  69. 'version': dist.version,
  70. 'location': dist.location,
  71. 'requires': [dep.project_name for dep in dist.requires()],
  72. 'required_by': get_requiring_packages(dist.project_name)
  73. }
  74. file_list = None
  75. metadata = ''
  76. if isinstance(dist, pkg_resources.DistInfoDistribution):
  77. # RECORDs should be part of .dist-info metadatas
  78. if dist.has_metadata('RECORD'):
  79. lines = dist.get_metadata_lines('RECORD')
  80. paths = [line.split(',')[0] for line in lines]
  81. paths = [os.path.join(dist.location, p) for p in paths]
  82. file_list = [os.path.relpath(p, dist.location) for p in paths]
  83. if dist.has_metadata('METADATA'):
  84. metadata = dist.get_metadata('METADATA')
  85. else:
  86. # Otherwise use pip's log for .egg-info's
  87. if dist.has_metadata('installed-files.txt'):
  88. paths = dist.get_metadata_lines('installed-files.txt')
  89. paths = [os.path.join(dist.egg_info, p) for p in paths]
  90. file_list = [os.path.relpath(p, dist.location) for p in paths]
  91. if dist.has_metadata('PKG-INFO'):
  92. metadata = dist.get_metadata('PKG-INFO')
  93. if dist.has_metadata('entry_points.txt'):
  94. entry_points = dist.get_metadata_lines('entry_points.txt')
  95. package['entry_points'] = entry_points
  96. if dist.has_metadata('INSTALLER'):
  97. for line in dist.get_metadata_lines('INSTALLER'):
  98. if line.strip():
  99. package['installer'] = line.strip()
  100. break
  101. # @todo: Should pkg_resources.Distribution have a
  102. # `get_pkg_info` method?
  103. feed_parser = FeedParser()
  104. feed_parser.feed(metadata)
  105. pkg_info_dict = feed_parser.close()
  106. for key in ('metadata-version', 'summary',
  107. 'home-page', 'author', 'author-email', 'license'):
  108. package[key] = pkg_info_dict.get(key)
  109. # It looks like FeedParser cannot deal with repeated headers
  110. classifiers = []
  111. for line in metadata.splitlines():
  112. if line.startswith('Classifier: '):
  113. classifiers.append(line[len('Classifier: '):])
  114. package['classifiers'] = classifiers
  115. if file_list:
  116. package['files'] = sorted(file_list)
  117. yield package
  118. def print_results(distributions, list_files=False, verbose=False):
  119. # type: (Iterator[Dict[str, str]], bool, bool) -> bool
  120. """
  121. Print the information from installed distributions found.
  122. """
  123. results_printed = False
  124. for i, dist in enumerate(distributions):
  125. results_printed = True
  126. if i > 0:
  127. write_output("---")
  128. write_output("Name: %s", dist.get('name', ''))
  129. write_output("Version: %s", dist.get('version', ''))
  130. write_output("Summary: %s", dist.get('summary', ''))
  131. write_output("Home-page: %s", dist.get('home-page', ''))
  132. write_output("Author: %s", dist.get('author', ''))
  133. write_output("Author-email: %s", dist.get('author-email', ''))
  134. write_output("License: %s", dist.get('license', ''))
  135. write_output("Location: %s", dist.get('location', ''))
  136. write_output("Requires: %s", ', '.join(dist.get('requires', [])))
  137. write_output("Required-by: %s", ', '.join(dist.get('required_by', [])))
  138. if verbose:
  139. write_output("Metadata-Version: %s",
  140. dist.get('metadata-version', ''))
  141. write_output("Installer: %s", dist.get('installer', ''))
  142. write_output("Classifiers:")
  143. for classifier in dist.get('classifiers', []):
  144. write_output(" %s", classifier)
  145. write_output("Entry-points:")
  146. for entry in dist.get('entry_points', []):
  147. write_output(" %s", entry.strip())
  148. if list_files:
  149. write_output("Files:")
  150. for line in dist.get('files', []):
  151. write_output(" %s", line.strip())
  152. if "files" not in dist:
  153. write_output("Cannot locate installed-files.txt")
  154. return results_printed