]>
Commit | Line | Data |
---|---|---|
53e6db90 DC |
1 | import glob |
2 | import os | |
3 | import subprocess | |
4 | import sys | |
5 | import tempfile | |
6 | import warnings | |
7 | from distutils import log | |
8 | from distutils.errors import DistutilsError | |
9 | ||
10 | import pkg_resources | |
11 | from setuptools.wheel import Wheel | |
12 | from ._deprecation_warning import SetuptoolsDeprecationWarning | |
13 | ||
14 | ||
15 | def _fixup_find_links(find_links): | |
16 | """Ensure find-links option end-up being a list of strings.""" | |
17 | if isinstance(find_links, str): | |
18 | return find_links.split() | |
19 | assert isinstance(find_links, (tuple, list)) | |
20 | return find_links | |
21 | ||
22 | ||
23 | def fetch_build_egg(dist, req): # noqa: C901 # is too complex (16) # FIXME | |
24 | """Fetch an egg needed for building. | |
25 | ||
26 | Use pip/wheel to fetch/build a wheel.""" | |
27 | warnings.warn( | |
28 | "setuptools.installer is deprecated. Requirements should " | |
29 | "be satisfied by a PEP 517 installer.", | |
30 | SetuptoolsDeprecationWarning, | |
31 | ) | |
32 | # Warn if wheel is not available | |
33 | try: | |
34 | pkg_resources.get_distribution('wheel') | |
35 | except pkg_resources.DistributionNotFound: | |
36 | dist.announce('WARNING: The wheel package is not available.', log.WARN) | |
37 | # Ignore environment markers; if supplied, it is required. | |
38 | req = strip_marker(req) | |
39 | # Take easy_install options into account, but do not override relevant | |
40 | # pip environment variables (like PIP_INDEX_URL or PIP_QUIET); they'll | |
41 | # take precedence. | |
42 | opts = dist.get_option_dict('easy_install') | |
43 | if 'allow_hosts' in opts: | |
44 | raise DistutilsError('the `allow-hosts` option is not supported ' | |
45 | 'when using pip to install requirements.') | |
46 | quiet = 'PIP_QUIET' not in os.environ and 'PIP_VERBOSE' not in os.environ | |
47 | if 'PIP_INDEX_URL' in os.environ: | |
48 | index_url = None | |
49 | elif 'index_url' in opts: | |
50 | index_url = opts['index_url'][1] | |
51 | else: | |
52 | index_url = None | |
53 | find_links = ( | |
54 | _fixup_find_links(opts['find_links'][1])[:] if 'find_links' in opts | |
55 | else [] | |
56 | ) | |
57 | if dist.dependency_links: | |
58 | find_links.extend(dist.dependency_links) | |
59 | eggs_dir = os.path.realpath(dist.get_egg_cache_dir()) | |
60 | environment = pkg_resources.Environment() | |
61 | for egg_dist in pkg_resources.find_distributions(eggs_dir): | |
62 | if egg_dist in req and environment.can_add(egg_dist): | |
63 | return egg_dist | |
64 | with tempfile.TemporaryDirectory() as tmpdir: | |
65 | cmd = [ | |
66 | sys.executable, '-m', 'pip', | |
67 | '--disable-pip-version-check', | |
68 | 'wheel', '--no-deps', | |
69 | '-w', tmpdir, | |
70 | ] | |
71 | if quiet: | |
72 | cmd.append('--quiet') | |
73 | if index_url is not None: | |
74 | cmd.extend(('--index-url', index_url)) | |
75 | for link in find_links or []: | |
76 | cmd.extend(('--find-links', link)) | |
77 | # If requirement is a PEP 508 direct URL, directly pass | |
78 | # the URL to pip, as `req @ url` does not work on the | |
79 | # command line. | |
80 | cmd.append(req.url or str(req)) | |
81 | try: | |
82 | subprocess.check_call(cmd) | |
83 | except subprocess.CalledProcessError as e: | |
84 | raise DistutilsError(str(e)) from e | |
85 | wheel = Wheel(glob.glob(os.path.join(tmpdir, '*.whl'))[0]) | |
86 | dist_location = os.path.join(eggs_dir, wheel.egg_name()) | |
87 | wheel.install_as_egg(dist_location) | |
88 | dist_metadata = pkg_resources.PathMetadata( | |
89 | dist_location, os.path.join(dist_location, 'EGG-INFO')) | |
90 | dist = pkg_resources.Distribution.from_filename( | |
91 | dist_location, metadata=dist_metadata) | |
92 | return dist | |
93 | ||
94 | ||
95 | def strip_marker(req): | |
96 | """ | |
97 | Return a new requirement without the environment marker to avoid | |
98 | calling pip with something like `babel; extra == "i18n"`, which | |
99 | would always be ignored. | |
100 | """ | |
101 | # create a copy to avoid mutating the input | |
102 | req = pkg_resources.Requirement.parse(str(req)) | |
103 | req.marker = None | |
104 | return req |