lock.py 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170
  1. import sys
  2. from optparse import Values
  3. from pathlib import Path
  4. from pip._internal.cache import WheelCache
  5. from pip._internal.cli import cmdoptions
  6. from pip._internal.cli.req_command import (
  7. RequirementCommand,
  8. with_cleanup,
  9. )
  10. from pip._internal.cli.status_codes import SUCCESS
  11. from pip._internal.models.pylock import Pylock, is_valid_pylock_file_name
  12. from pip._internal.operations.build.build_tracker import get_build_tracker
  13. from pip._internal.req.req_install import (
  14. check_legacy_setup_py_options,
  15. )
  16. from pip._internal.utils.logging import getLogger
  17. from pip._internal.utils.misc import (
  18. get_pip_version,
  19. )
  20. from pip._internal.utils.temp_dir import TempDirectory
  21. logger = getLogger(__name__)
  22. class LockCommand(RequirementCommand):
  23. """
  24. EXPERIMENTAL - Lock packages and their dependencies from:
  25. - PyPI (and other indexes) using requirement specifiers.
  26. - VCS project urls.
  27. - Local project directories.
  28. - Local or remote source archives.
  29. pip also supports locking from "requirements files", which provide an easy
  30. way to specify a whole environment to be installed.
  31. The generated lock file is only guaranteed to be valid for the current
  32. python version and platform.
  33. """
  34. usage = """
  35. %prog [options] [-e] <local project path> ...
  36. %prog [options] <requirement specifier> [package-index-options] ...
  37. %prog [options] -r <requirements file> [package-index-options] ...
  38. %prog [options] <archive url/path> ..."""
  39. def add_options(self) -> None:
  40. self.cmd_opts.add_option(
  41. cmdoptions.PipOption(
  42. "--output",
  43. "-o",
  44. dest="output_file",
  45. metavar="path",
  46. type="path",
  47. default="pylock.toml",
  48. help="Lock file name (default=pylock.toml). Use - for stdout.",
  49. )
  50. )
  51. self.cmd_opts.add_option(cmdoptions.requirements())
  52. self.cmd_opts.add_option(cmdoptions.constraints())
  53. self.cmd_opts.add_option(cmdoptions.no_deps())
  54. self.cmd_opts.add_option(cmdoptions.pre())
  55. self.cmd_opts.add_option(cmdoptions.editable())
  56. self.cmd_opts.add_option(cmdoptions.src())
  57. self.cmd_opts.add_option(cmdoptions.ignore_requires_python())
  58. self.cmd_opts.add_option(cmdoptions.no_build_isolation())
  59. self.cmd_opts.add_option(cmdoptions.use_pep517())
  60. self.cmd_opts.add_option(cmdoptions.no_use_pep517())
  61. self.cmd_opts.add_option(cmdoptions.check_build_deps())
  62. self.cmd_opts.add_option(cmdoptions.config_settings())
  63. self.cmd_opts.add_option(cmdoptions.no_binary())
  64. self.cmd_opts.add_option(cmdoptions.only_binary())
  65. self.cmd_opts.add_option(cmdoptions.prefer_binary())
  66. self.cmd_opts.add_option(cmdoptions.require_hashes())
  67. self.cmd_opts.add_option(cmdoptions.progress_bar())
  68. index_opts = cmdoptions.make_option_group(
  69. cmdoptions.index_group,
  70. self.parser,
  71. )
  72. self.parser.insert_option_group(0, index_opts)
  73. self.parser.insert_option_group(0, self.cmd_opts)
  74. @with_cleanup
  75. def run(self, options: Values, args: list[str]) -> int:
  76. logger.verbose("Using %s", get_pip_version())
  77. logger.warning(
  78. "pip lock is currently an experimental command. "
  79. "It may be removed/changed in a future release "
  80. "without prior warning."
  81. )
  82. session = self.get_default_session(options)
  83. finder = self._build_package_finder(
  84. options=options,
  85. session=session,
  86. ignore_requires_python=options.ignore_requires_python,
  87. )
  88. build_tracker = self.enter_context(get_build_tracker())
  89. directory = TempDirectory(
  90. delete=not options.no_clean,
  91. kind="install",
  92. globally_managed=True,
  93. )
  94. reqs = self.get_requirements(args, options, finder, session)
  95. check_legacy_setup_py_options(options, reqs)
  96. wheel_cache = WheelCache(options.cache_dir)
  97. # Only when installing is it permitted to use PEP 660.
  98. # In other circumstances (pip wheel, pip download) we generate
  99. # regular (i.e. non editable) metadata and wheels.
  100. for req in reqs:
  101. req.permit_editable_wheels = True
  102. preparer = self.make_requirement_preparer(
  103. temp_build_dir=directory,
  104. options=options,
  105. build_tracker=build_tracker,
  106. session=session,
  107. finder=finder,
  108. use_user_site=False,
  109. verbosity=self.verbosity,
  110. )
  111. resolver = self.make_resolver(
  112. preparer=preparer,
  113. finder=finder,
  114. options=options,
  115. wheel_cache=wheel_cache,
  116. use_user_site=False,
  117. ignore_installed=True,
  118. ignore_requires_python=options.ignore_requires_python,
  119. upgrade_strategy="to-satisfy-only",
  120. use_pep517=options.use_pep517,
  121. )
  122. self.trace_basic_info(finder)
  123. requirement_set = resolver.resolve(reqs, check_supported_wheels=True)
  124. if options.output_file == "-":
  125. base_dir = Path.cwd()
  126. else:
  127. output_file_path = Path(options.output_file)
  128. if not is_valid_pylock_file_name(output_file_path):
  129. logger.warning(
  130. "%s is not a valid lock file name.",
  131. output_file_path,
  132. )
  133. base_dir = output_file_path.parent
  134. pylock_toml = Pylock.from_install_requirements(
  135. requirement_set.requirements.values(), base_dir=base_dir
  136. ).as_toml()
  137. if options.output_file == "-":
  138. sys.stdout.write(pylock_toml)
  139. else:
  140. output_file_path.write_text(pylock_toml, encoding="utf-8")
  141. return SUCCESS