Skip to content

Commit 7946c25

Browse files
committed
move extras reading into a function
1 parent bb85b96 commit 7946c25

File tree

1 file changed

+12
-9
lines changed

1 file changed

+12
-9
lines changed

setup.py

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -23,21 +23,24 @@ def read(*names, **kwargs):
2323
return fh.read()
2424

2525

26+
def read_extras(*prefixes):
27+
requirements = []
28+
extras = {}
29+
30+
for prefix in prefixes:
31+
requirements.extend(read("{}-requirements.in".format(prefix)).splitlines())
32+
extras[prefix] = requirements.copy()
33+
34+
return extras
35+
36+
2637
try:
2738
with open("requirements.in") as f:
2839
INSTALL_REQUIRES = f.read().splitlines()
2940
except FileNotFoundError:
3041
print(sys.exc_info())
3142
INSTALL_REQUIRES = []
3243

33-
EXTRAS_PREFIXES = ["test", "dev"]
34-
35-
requirements = []
36-
extras = {}
37-
for prefix in EXTRAS_PREFIXES:
38-
requirements.extend(read("{}-requirements.in".format(prefix)).splitlines())
39-
extras[prefix] = requirements.copy()
40-
4144
ns = {}
4245
exec(read("src/desert/_version.py"), ns)
4346
VERSION = ns["__version__"]
@@ -88,5 +91,5 @@ def read(*names, **kwargs):
8891
install_requires=INSTALL_REQUIRES
8992
# eg: "aspectlib==1.1.1", "six>=1.7",
9093
,
91-
extras_require=extras,
94+
extras_require=read_extras('test', 'dev'),
9295
)

0 commit comments

Comments
 (0)