Update update-sanity-requirements.py script (#81424)
Frozen requirements can now preserve any explicitly installed package that would normally be omitted, not just setuptools.
This commit is contained in:
parent
2ebeb43d6f
commit
dbb3feddaf
|
@ -15,6 +15,7 @@ import venv
|
|||
|
||||
import packaging.version
|
||||
import packaging.specifiers
|
||||
import packaging.requirements
|
||||
|
||||
try:
|
||||
import argcomplete
|
||||
|
@ -34,6 +35,11 @@ class SanityTest:
|
|||
source_path: pathlib.Path
|
||||
|
||||
def freeze_requirements(self) -> None:
|
||||
source_requirements = [packaging.requirements.Requirement(re.sub(' #.*$', '', line)) for line in self.source_path.read_text().splitlines()]
|
||||
|
||||
install_packages = {requirement.name for requirement in source_requirements}
|
||||
exclude_packages = {'distribute', 'pip', 'setuptools', 'wheel'} - install_packages
|
||||
|
||||
with tempfile.TemporaryDirectory() as venv_dir:
|
||||
venv.create(venv_dir, with_pip=True)
|
||||
|
||||
|
@ -49,13 +55,6 @@ class SanityTest:
|
|||
subprocess.run(pip + ['install', 'wheel'], env=env, check=True) # make bdist_wheel available during pip install
|
||||
subprocess.run(pip + ['install', '-r', self.source_path], env=env, check=True)
|
||||
|
||||
keep_setuptools = any(line.startswith('setuptools ') for line in self.source_path.read_text().splitlines())
|
||||
|
||||
exclude_packages = ['pip', 'distribute', 'wheel']
|
||||
|
||||
if not keep_setuptools:
|
||||
exclude_packages.append('setuptools')
|
||||
|
||||
freeze_options = ['--all']
|
||||
|
||||
for exclude_package in exclude_packages:
|
||||
|
|
Loading…
Reference in New Issue