Tipragot
628be439b8
Cela permet de ne pas avoir de problèmes de compatibilité car python est dans le git.
129 lines
4.1 KiB
Python
129 lines
4.1 KiB
Python
"""Helper code used to generate ``requires.txt`` files in the egg-info directory.
|
|
|
|
The ``requires.txt`` file has an specific format:
|
|
- Environment markers need to be part of the section headers and
|
|
should not be part of the requirement spec itself.
|
|
|
|
See https://setuptools.pypa.io/en/latest/deprecated/python_eggs.html#requires-txt
|
|
"""
|
|
import io
|
|
from collections import defaultdict
|
|
from itertools import filterfalse
|
|
from typing import Dict, List, Tuple, Mapping, TypeVar
|
|
|
|
from .. import _reqs
|
|
from ..extern.jaraco.text import yield_lines
|
|
from ..extern.packaging.requirements import Requirement
|
|
|
|
|
|
# dict can work as an ordered set
|
|
_T = TypeVar("_T")
|
|
_Ordered = Dict[_T, None]
|
|
_ordered = dict
|
|
_StrOrIter = _reqs._StrOrIter
|
|
|
|
|
|
def _prepare(
|
|
install_requires: _StrOrIter, extras_require: Mapping[str, _StrOrIter]
|
|
) -> Tuple[List[str], Dict[str, List[str]]]:
|
|
"""Given values for ``install_requires`` and ``extras_require``
|
|
create modified versions in a way that can be written in ``requires.txt``
|
|
"""
|
|
extras = _convert_extras_requirements(extras_require)
|
|
return _move_install_requirements_markers(install_requires, extras)
|
|
|
|
|
|
def _convert_extras_requirements(
|
|
extras_require: _StrOrIter,
|
|
) -> Mapping[str, _Ordered[Requirement]]:
|
|
"""
|
|
Convert requirements in `extras_require` of the form
|
|
`"extra": ["barbazquux; {marker}"]` to
|
|
`"extra:{marker}": ["barbazquux"]`.
|
|
"""
|
|
output: Mapping[str, _Ordered[Requirement]] = defaultdict(dict)
|
|
for section, v in extras_require.items():
|
|
# Do not strip empty sections.
|
|
output[section]
|
|
for r in _reqs.parse(v):
|
|
output[section + _suffix_for(r)].setdefault(r)
|
|
|
|
return output
|
|
|
|
|
|
def _move_install_requirements_markers(
|
|
install_requires: _StrOrIter, extras_require: Mapping[str, _Ordered[Requirement]]
|
|
) -> Tuple[List[str], Dict[str, List[str]]]:
|
|
"""
|
|
The ``requires.txt`` file has an specific format:
|
|
- Environment markers need to be part of the section headers and
|
|
should not be part of the requirement spec itself.
|
|
|
|
Move requirements in ``install_requires`` that are using environment
|
|
markers ``extras_require``.
|
|
"""
|
|
|
|
# divide the install_requires into two sets, simple ones still
|
|
# handled by install_requires and more complex ones handled by extras_require.
|
|
|
|
inst_reqs = list(_reqs.parse(install_requires))
|
|
simple_reqs = filter(_no_marker, inst_reqs)
|
|
complex_reqs = filterfalse(_no_marker, inst_reqs)
|
|
simple_install_requires = list(map(str, simple_reqs))
|
|
|
|
for r in complex_reqs:
|
|
extras_require[':' + str(r.marker)].setdefault(r)
|
|
|
|
expanded_extras = dict(
|
|
# list(dict.fromkeys(...)) ensures a list of unique strings
|
|
(k, list(dict.fromkeys(str(r) for r in map(_clean_req, v))))
|
|
for k, v in extras_require.items()
|
|
)
|
|
|
|
return simple_install_requires, expanded_extras
|
|
|
|
|
|
def _suffix_for(req):
|
|
"""Return the 'extras_require' suffix for a given requirement."""
|
|
return ':' + str(req.marker) if req.marker else ''
|
|
|
|
|
|
def _clean_req(req):
|
|
"""Given a Requirement, remove environment markers and return it"""
|
|
r = Requirement(str(req)) # create a copy before modifying
|
|
r.marker = None
|
|
return r
|
|
|
|
|
|
def _no_marker(req):
|
|
return not req.marker
|
|
|
|
|
|
def _write_requirements(stream, reqs):
|
|
lines = yield_lines(reqs or ())
|
|
|
|
def append_cr(line):
|
|
return line + '\n'
|
|
|
|
lines = map(append_cr, lines)
|
|
stream.writelines(lines)
|
|
|
|
|
|
def write_requirements(cmd, basename, filename):
|
|
dist = cmd.distribution
|
|
data = io.StringIO()
|
|
install_requires, extras_require = _prepare(
|
|
dist.install_requires or (), dist.extras_require or {}
|
|
)
|
|
_write_requirements(data, install_requires)
|
|
for extra in sorted(extras_require):
|
|
data.write('\n[{extra}]\n'.format(**vars()))
|
|
_write_requirements(data, extras_require[extra])
|
|
cmd.write_or_delete_file("requirements", filename, data.getvalue())
|
|
|
|
|
|
def write_setup_requirements(cmd, basename, filename):
|
|
data = io.StringIO()
|
|
_write_requirements(data, cmd.distribution.setup_requires)
|
|
cmd.write_or_delete_file("setup-requirements", filename, data.getvalue())
|