download.py 5.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146
  1. import logging
  2. import os
  3. from optparse import Values
  4. from pip._internal.cli import cmdoptions
  5. from pip._internal.cli.cmdoptions import make_target_python
  6. from pip._internal.cli.req_command import RequirementCommand, with_cleanup
  7. from pip._internal.cli.status_codes import SUCCESS
  8. from pip._internal.operations.build.build_tracker import get_build_tracker
  9. from pip._internal.utils.misc import ensure_dir, normalize_path, write_output
  10. from pip._internal.utils.temp_dir import TempDirectory
  11. logger = logging.getLogger(__name__)
  12. class DownloadCommand(RequirementCommand):
  13. """
  14. Download packages from:
  15. - PyPI (and other indexes) using requirement specifiers.
  16. - VCS project urls.
  17. - Local project directories.
  18. - Local or remote source archives.
  19. pip also supports downloading from "requirements files", which provide
  20. an easy way to specify a whole environment to be downloaded.
  21. """
  22. usage = """
  23. %prog [options] <requirement specifier> [package-index-options] ...
  24. %prog [options] -r <requirements file> [package-index-options] ...
  25. %prog [options] <vcs project url> ...
  26. %prog [options] <local project path> ...
  27. %prog [options] <archive url/path> ..."""
  28. def add_options(self) -> None:
  29. self.cmd_opts.add_option(cmdoptions.constraints())
  30. self.cmd_opts.add_option(cmdoptions.build_constraints())
  31. self.cmd_opts.add_option(cmdoptions.requirements())
  32. self.cmd_opts.add_option(cmdoptions.requirements_from_scripts())
  33. self.cmd_opts.add_option(cmdoptions.no_deps())
  34. self.cmd_opts.add_option(cmdoptions.src())
  35. self.cmd_opts.add_option(cmdoptions.require_hashes())
  36. self.cmd_opts.add_option(cmdoptions.progress_bar())
  37. self.cmd_opts.add_option(cmdoptions.no_build_isolation())
  38. self.cmd_opts.add_option(cmdoptions.use_pep517())
  39. self.cmd_opts.add_option(cmdoptions.check_build_deps())
  40. self.cmd_opts.add_option(cmdoptions.ignore_requires_python())
  41. self.cmd_opts.add_option(
  42. "-d",
  43. "--dest",
  44. "--destination-dir",
  45. "--destination-directory",
  46. dest="download_dir",
  47. metavar="dir",
  48. default=os.curdir,
  49. help="Download packages into <dir>.",
  50. )
  51. cmdoptions.add_target_python_options(self.cmd_opts)
  52. index_opts = cmdoptions.make_option_group(
  53. cmdoptions.index_group,
  54. self.parser,
  55. )
  56. selection_opts = cmdoptions.make_option_group(
  57. cmdoptions.package_selection_group,
  58. self.parser,
  59. )
  60. self.parser.insert_option_group(0, index_opts)
  61. self.parser.insert_option_group(0, selection_opts)
  62. self.parser.insert_option_group(0, self.cmd_opts)
  63. @with_cleanup
  64. def run(self, options: Values, args: list[str]) -> int:
  65. options.ignore_installed = True
  66. # editable doesn't really make sense for `pip download`, but the bowels
  67. # of the RequirementSet code require that property.
  68. options.editables = []
  69. cmdoptions.check_dist_restriction(options)
  70. cmdoptions.check_build_constraints(options)
  71. cmdoptions.check_release_control_exclusive(options)
  72. options.download_dir = normalize_path(options.download_dir)
  73. ensure_dir(options.download_dir)
  74. session = self.get_default_session(options)
  75. target_python = make_target_python(options)
  76. finder = self._build_package_finder(
  77. options=options,
  78. session=session,
  79. target_python=target_python,
  80. ignore_requires_python=options.ignore_requires_python,
  81. )
  82. build_tracker = self.enter_context(get_build_tracker())
  83. directory = TempDirectory(
  84. delete=not options.no_clean,
  85. kind="download",
  86. globally_managed=True,
  87. )
  88. reqs = self.get_requirements(args, options, finder, session)
  89. preparer = self.make_requirement_preparer(
  90. temp_build_dir=directory,
  91. options=options,
  92. build_tracker=build_tracker,
  93. session=session,
  94. finder=finder,
  95. download_dir=options.download_dir,
  96. use_user_site=False,
  97. verbosity=self.verbosity,
  98. )
  99. resolver = self.make_resolver(
  100. preparer=preparer,
  101. finder=finder,
  102. options=options,
  103. ignore_requires_python=options.ignore_requires_python,
  104. py_version_info=options.python_version,
  105. )
  106. self.trace_basic_info(finder)
  107. requirement_set = resolver.resolve(reqs, check_supported_wheels=True)
  108. preparer.prepare_linked_requirements_more(requirement_set.requirements.values())
  109. downloaded: list[str] = []
  110. for req in requirement_set.requirements.values():
  111. if req.satisfied_by is None:
  112. assert req.name is not None
  113. preparer.save_linked_requirement(req)
  114. downloaded.append(req.name)
  115. if downloaded:
  116. write_output("Successfully downloaded %s", " ".join(downloaded))
  117. return SUCCESS