Spaces:
Building
Building
| import glob | |
| import os | |
| import subprocess | |
| import sys | |
| import tempfile | |
| from distutils import log | |
| from distutils.errors import DistutilsError | |
| import pkg_resources | |
| from setuptools.wheel import Wheel | |
| def _fixup_find_links(find_links): | |
| """Ensure find-links option end-up being a list of strings.""" | |
| if isinstance(find_links, str): | |
| return find_links.split() | |
| assert isinstance(find_links, (tuple, list)) | |
| return find_links | |
| def fetch_build_egg(dist, req): # noqa: C901 # is too complex (16) # FIXME | |
| """Fetch an egg needed for building. | |
| Use pip/wheel to fetch/build a wheel.""" | |
| # Warn if wheel is not available | |
| try: | |
| pkg_resources.get_distribution('wheel') | |
| except pkg_resources.DistributionNotFound: | |
| dist.announce('WARNING: The wheel package is not available.', log.WARN) | |
| # Ignore environment markers; if supplied, it is required. | |
| req = strip_marker(req) | |
| # Take easy_install options into account, but do not override relevant | |
| # pip environment variables (like PIP_INDEX_URL or PIP_QUIET); they'll | |
| # take precedence. | |
| opts = dist.get_option_dict('easy_install') | |
| if 'allow_hosts' in opts: | |
| raise DistutilsError('the `allow-hosts` option is not supported ' | |
| 'when using pip to install requirements.') | |
| quiet = 'PIP_QUIET' not in os.environ and 'PIP_VERBOSE' not in os.environ | |
| if 'PIP_INDEX_URL' in os.environ: | |
| index_url = None | |
| elif 'index_url' in opts: | |
| index_url = opts['index_url'][1] | |
| else: | |
| index_url = None | |
| find_links = ( | |
| _fixup_find_links(opts['find_links'][1])[:] if 'find_links' in opts | |
| else [] | |
| ) | |
| if dist.dependency_links: | |
| find_links.extend(dist.dependency_links) | |
| eggs_dir = os.path.realpath(dist.get_egg_cache_dir()) | |
| environment = pkg_resources.Environment() | |
| for egg_dist in pkg_resources.find_distributions(eggs_dir): | |
| if egg_dist in req and environment.can_add(egg_dist): | |
| return egg_dist | |
| with tempfile.TemporaryDirectory() as tmpdir: | |
| cmd = [ | |
| sys.executable, '-m', 'pip', | |
| '--disable-pip-version-check', | |
| 'wheel', '--no-deps', | |
| '-w', tmpdir, | |
| ] | |
| if quiet: | |
| cmd.append('--quiet') | |
| if index_url is not None: | |
| cmd.extend(('--index-url', index_url)) | |
| for link in find_links or []: | |
| cmd.extend(('--find-links', link)) | |
| # If requirement is a PEP 508 direct URL, directly pass | |
| # the URL to pip, as `req @ url` does not work on the | |
| # command line. | |
| cmd.append(req.url or str(req)) | |
| try: | |
| subprocess.check_call(cmd) | |
| except subprocess.CalledProcessError as e: | |
| raise DistutilsError(str(e)) from e | |
| wheel = Wheel(glob.glob(os.path.join(tmpdir, '*.whl'))[0]) | |
| dist_location = os.path.join(eggs_dir, wheel.egg_name()) | |
| wheel.install_as_egg(dist_location) | |
| dist_metadata = pkg_resources.PathMetadata( | |
| dist_location, os.path.join(dist_location, 'EGG-INFO')) | |
| dist = pkg_resources.Distribution.from_filename( | |
| dist_location, metadata=dist_metadata) | |
| return dist | |
| def strip_marker(req): | |
| """ | |
| Return a new requirement without the environment marker to avoid | |
| calling pip with something like `babel; extra == "i18n"`, which | |
| would always be ignored. | |
| """ | |
| # create a copy to avoid mutating the input | |
| req = pkg_resources.Requirement.parse(str(req)) | |
| req.marker = None | |
| return req | |