-
Notifications
You must be signed in to change notification settings - Fork 20
/
check_dependencies.py
51 lines (42 loc) · 1.31 KB
/
check_dependencies.py
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
#!python
'''
Adapted from https://github.com/scikit-image/skimage-tutorials/check_setup.py (ade46c2)
The goal of this file is to give users a clean and descriptive check of their environments.
'''
import sys
import os
from distutils.version import LooseVersion
with open(os.path.join(os.path.dirname(__file__), 'requirements.txt')) as f:
reqs = f.readlines()
#used for renaming packages
pkg_names = { }
for pkg_str in reqs:
pkg_split = pkg_str.strip().split()
if len(pkg_split)>1:
pkg,eq,version = pkg_split
else:
pkg = pkg_split[0]
version = '0.0.0'
module_name = pkg_names.get(pkg, pkg)
try:
m = __import__(module_name)
status = '✓'
except ImportError as e:
m = None
if (pkg != 'numpy' and 'numpy' in str(e)):
status = '?'
version_installed = 'Needs NumPy'
else:
version_installed = 'Not installed'
status = 'X'
if m is not None:
version_installed = m.__version__
if LooseVersion(version) > LooseVersion(version_installed):
status = 'X'
if 'sphinx' in pkg:
reqstring = '(only for building docs)'
else:
reqstring = ''
print('[{}] {:<20} {:20} {}'.format(
status, pkg.ljust(13), version_installed, reqstring)
)