9
9
#
10
10
"""Generate requirements from `setup.py` and `requirements-devel.txt`."""
11
11
12
+ import ast
12
13
import os
13
14
import sys
14
- import ast
15
15
16
16
try :
17
17
import configparser
18
18
except ImportError : # pragma: no cover
19
19
import ConfigParser as configparser
20
20
21
+ import distutils .core
22
+
21
23
import mock
22
24
import setuptools
23
- import distutils .core
24
25
from commoncode .command import pushd
25
26
from packvers .requirements import Requirement
26
27
27
28
28
29
def minver_error (pkg_name ):
29
30
"""Report error about missing minimum version constraint and exit."""
30
31
print (
31
- 'ERROR: specify minimal version of "{0}" using '
32
- '">=" or "=="' .format (pkg_name ),
32
+ 'ERROR: specify minimal version of "{0}" using ' '">=" or "=="' .format (pkg_name ),
33
33
file = sys .stderr ,
34
34
)
35
35
sys .exit (1 )
@@ -68,9 +68,7 @@ def iter_requirements(level, extras, setup_file):
68
68
asnames [(n .asname if n .asname is not None else n .name )] = n .name
69
69
for elem in ast .walk (node ):
70
70
# for function imports, e.g. from setuptools import setup; setup()
71
- if isinstance (elem , ast .ImportFrom ) and "setup" in [
72
- e .name for e in elem .names
73
- ]:
71
+ if isinstance (elem , ast .ImportFrom ) and "setup" in [e .name for e in elem .names ]:
74
72
imports .append (elem .module )
75
73
# for module imports, e.g. import setuptools; setuptools.setup(...)
76
74
elif (
@@ -93,16 +91,12 @@ def iter_requirements(level, extras, setup_file):
93
91
and elem .value .func .attr == "setup"
94
92
):
95
93
name = (
96
- str (elem .value .func .value .value .id )
97
- + "."
98
- + str (elem .value .func .value .attr )
94
+ str (elem .value .func .value .value .id ) + "." + str (elem .value .func .value .attr )
99
95
)
100
96
if name in asnames .keys ():
101
97
name = asnames [name ]
102
98
imports .append (name )
103
- setup_providers = [
104
- i for i in imports if i in ["distutils.core" , "setuptools" ]
105
- ]
99
+ setup_providers = [i for i in imports if i in ["distutils.core" , "setuptools" ]]
106
100
if len (setup_providers ) == 0 :
107
101
print (
108
102
f"Warning: unable to recognize setup provider in { setup_file } : "
@@ -195,9 +189,7 @@ def iter_requirements(level, extras, setup_file):
195
189
result [pkg .name ] = "{0}=={1}" .format (build_pkg_name (pkg ), specs ["~=" ])
196
190
else :
197
191
ver , _ = os .path .splitext (specs ["~=" ])
198
- result [pkg .name ] = "{0}>={1},=={2}.*" .format (
199
- build_pkg_name (pkg ), specs ["~=" ], ver
200
- )
192
+ result [pkg .name ] = "{0}>={1},=={2}.*" .format (build_pkg_name (pkg ), specs ["~=" ], ver )
201
193
202
194
else :
203
195
if level == "min" :
0 commit comments