| |
@@ -16,7 +16,7 @@
|
| |
from distutils.sysconfig import get_python_lib
|
| |
from os.path import dirname, sep
|
| |
import re
|
| |
- from sys import argv, stdin, stderr
|
| |
+ from sys import argv, stdin, stderr, version_info
|
| |
from warnings import warn
|
| |
|
| |
from packaging.requirements import Requirement as Requirement_
|
| |
@@ -44,8 +44,12 @@
|
| |
|
| |
try:
|
| |
from importlib.metadata import PathDistribution
|
| |
+ from importlib.metadata import PackageNotFoundError
|
| |
+ from importlib.metadata import version as version_of
|
| |
except ImportError:
|
| |
from importlib_metadata import PathDistribution
|
| |
+ from importlib_metadata import PackageNotFoundError
|
| |
+ from importlib_metadata import version as version_of
|
| |
|
| |
try:
|
| |
from pathlib import Path
|
| |
@@ -252,6 +256,67 @@
|
| |
"extra": extra}
|
| |
|
| |
|
| |
+ def find_setuptools_version(py_version):
|
| |
+ """Figure out setuptools version installed for the given Python X.Y version"""
|
| |
+
|
| |
+ # first, do it the sane way, if we run that Python version
|
| |
+ # nb: this breaks if we run on PyPy and generate for CPython or vice versa
|
| |
+ if '{}.{}'.format(*version_info[:2]) == py_version:
|
| |
+ try:
|
| |
+ return parse(version_of('setuptools'))
|
| |
+ except PackageNotFoundError:
|
| |
+ pass
|
| |
+
|
| |
+ # next, run the Python version and import setuptools from it
|
| |
+ # nb: this also breaks if we run on different implementation
|
| |
+ try:
|
| |
+ import subprocess
|
| |
+ code = 'import setuptools; print(setuptools.__version__)'
|
| |
+ output = subprocess.check_output(['python{}'.format(dist.py_version), '-Esc', code],
|
| |
+ universal_newlines=True)
|
| |
+ return parse(output.strip())
|
| |
+ except Exception: # anything can go wrong here
|
| |
+ pass
|
| |
+
|
| |
+ warn('Cannot check setuptools version for Python {}; '
|
| |
+ 'assuming an old version for safety.'.format(py_version),
|
| |
+ RuntimeWarning)
|
| |
+ return parse('0')
|
| |
+
|
| |
+
|
| |
+ def console_scripts_deps(dist, metadirname):
|
| |
+ if metadirname.endswith('.dist-info'):
|
| |
+ # console scripts from .dist-info have no extra deps
|
| |
+ return []
|
| |
+
|
| |
+ groups = {ep.group for ep in dist.entry_points}
|
| |
+ if not {"console_scripts", "gui_scripts"} & groups:
|
| |
+ # a different kind of entrypoint
|
| |
+ return []
|
| |
+
|
| |
+ python_version = parse(dist.py_version)
|
| |
+ setuptools_version = find_setuptools_version(dist.py_version)
|
| |
+
|
| |
+ # https://setuptools.readthedocs.io/en/latest/history.html#v47-3-0
|
| |
+ if setuptools_version >= parse('47.3.0'):
|
| |
+ if python_version < parse('3.6'):
|
| |
+ # don't use importlib_metadata on very old Pythons,
|
| |
+ # use pkg_resources from setuptools
|
| |
+ return [Requirement('setuptools')]
|
| |
+ if python_version < parse('3.8'):
|
| |
+ # technically this can also fallback to pkg_resources,
|
| |
+ # but we prefer the lighter requirement
|
| |
+ return [Requirement('importlib_metadata')]
|
| |
+ return []
|
| |
+
|
| |
+ # https://setuptools.readthedocs.io/en/latest/history.html#v47-2-0
|
| |
+ if setuptools_version >= parse('47.2.0') and python_version >= parse('3.8'):
|
| |
+ return []
|
| |
+
|
| |
+ # older versions use pkg_resources from setuptools
|
| |
+ return [Requirement('setuptools')]
|
| |
+
|
| |
+
|
| |
if __name__ == "__main__":
|
| |
"""To allow this script to be importable (and its classes/functions
|
| |
reused), actions are performed only when run as a main script."""
|
| |
@@ -438,15 +503,12 @@
|
| |
else:
|
| |
deps = dist.requirements
|
| |
|
| |
- # console_scripts/gui_scripts entry points need pkg_resources from setuptools
|
| |
- if (dist.entry_points and
|
| |
- (lower.endswith('.egg') or
|
| |
- lower.endswith('.egg-info'))):
|
| |
- groups = {ep.group for ep in dist.entry_points}
|
| |
- if {"console_scripts", "gui_scripts"} & groups:
|
| |
- # stick them first so any more specific requirement
|
| |
- # overrides it
|
| |
- deps.insert(0, Requirement('setuptools'))
|
| |
+ # console_scripts/gui_scripts entry points might need extra dependencies
|
| |
+ # we insert the requirement to the beginning of the list
|
| |
+ # so any more specific requirement on the same package can override it
|
| |
+ if dist.entry_points:
|
| |
+ deps = console_scripts_deps(dist, lower) + deps
|
| |
+
|
| |
# add requires/recommends based on egg/dist metadata
|
| |
for dep in deps:
|
| |
# Even if we're requiring `foo[bar]`, also require `foo`
|
| |
torsava commented 3 years ago
Minor nitpick: Why each import on it's own line when other imports use a multi-import style?