1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
"""A set of functions helping generating pip constraint files."""
from __future__ import print_function # noqa: WPS422
import functools
import os
import platform
import subprocess # noqa: S404
import sys
PYTHON_IMPLEMENTATION_MAP = { # noqa: WPS407
'cpython': 'cp',
'ironpython': 'ip',
'jython': 'jy',
'python': 'py',
'pypy': 'pp',
}
PYTHON_IMPLEMENTATION = platform.python_implementation()
print_info = functools.partial(print, file=sys.stderr)
def get_runtime_python_tag():
"""Identify the Python tag of the current runtime.
:returns: Python tag.
"""
python_minor_ver = sys.version_info[:2]
try:
sys_impl = sys.implementation.name
except AttributeError:
sys_impl = PYTHON_IMPLEMENTATION.lower()
# pylint: disable=possibly-unused-variable
python_tag_prefix = PYTHON_IMPLEMENTATION_MAP.get(sys_impl, sys_impl)
# pylint: disable=possibly-unused-variable
python_minor_ver_tag = ''.join(map(str, python_minor_ver))
return (
'{python_tag_prefix!s}{python_minor_ver_tag!s}'.
format(**locals()) # noqa: WPS421
)
def get_constraint_file_path(req_dir, toxenv, python_tag):
"""Identify the constraints filename for the current environment.
:param req_dir: Requirements directory.
:param toxenv: tox testenv.
:param python_tag: Python tag.
:returns: Constraints filename for the current environment.
"""
sys_platform = sys.platform
# pylint: disable=possibly-unused-variable
platform_machine = platform.machine().lower()
if toxenv in {'py', 'python'}:
extra_prefix = 'py' if PYTHON_IMPLEMENTATION == 'PyPy' else ''
toxenv = '{prefix}py{ver}'.format(
prefix=extra_prefix,
ver=python_tag[2:],
)
if sys_platform == 'linux2':
sys_platform = 'linux'
constraint_name = (
'tox-{toxenv}-{python_tag}-{sys_platform}-{platform_machine}'.
format(**locals()) # noqa: WPS421
)
return os.path.join(req_dir, os.path.extsep.join((constraint_name, 'txt')))
def make_pip_cmd(pip_args, constraint_file_path):
"""Inject a lockfile constraint into the pip command if present.
:param pip_args: pip arguments.
:param constraint_file_path: Path to a ``constraints.txt``-compatible file.
:returns: pip command.
"""
pip_cmd = [sys.executable, '-m', 'pip'] + pip_args
if os.path.isfile(constraint_file_path):
pip_cmd += ['--constraint', constraint_file_path]
else:
print_info(
'WARNING: The expected pinned constraints file for the current '
'env does not exist (should be "{constraint_file_path}").'.
format(**locals()), # noqa: WPS421
)
return pip_cmd
def run_cmd(cmd):
"""Invoke a shell command after logging it.
:param cmd: The command to invoke.
"""
print_info(
'Invoking the following command: {cmd}'.
format(cmd=' '.join(cmd)),
)
subprocess.check_call(cmd) # noqa: S603
|
