-
Notifications
You must be signed in to change notification settings - Fork 52
/
setup.py
53 lines (46 loc) · 2.29 KB
/
setup.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
52
53
# -*- coding: utf-8 -*-
from ast import parse
from distutils.sysconfig import get_python_lib
from functools import partial
from os import path, listdir
from platform import python_version_tuple
from setuptools import setup, find_packages
if python_version_tuple()[0] == '3':
imap = map
ifilter = filter
else:
from itertools import imap, ifilter
if __name__ == '__main__':
package_name = 'mnet_deep_cdr'
with open(path.join(package_name, '__init__.py')) as f:
__author__, __version__ = imap(
lambda buf: next(imap(lambda e: e.value.s, parse(buf).body)),
ifilter(lambda line: line.startswith('__version__') or line.startswith('__author__'), f)
)
to_funcs = lambda *paths: (partial(path.join, path.dirname(__file__), package_name, *paths),
partial(path.join, get_python_lib(prefix=''), package_name, *paths))
_data_join, _data_install_dir = to_funcs('_data')
ipynb_checkpoints_join, ipynb_checkpoints_install_dir = to_funcs('.ipynb_checkpoints')
deep_model_join, deep_model_install_dir = to_funcs('deep_model')
# mat_scr_join, mat_scr_install_dir = to_funcs('mat_scr')
REFUGE_result_join, REFUGE_result_install_dir = to_funcs('REFUGE_result')
result_join, result_install_dir = to_funcs('result')
test_img_join, test_img_install_dir = to_funcs('test_img')
setup(
name=package_name,
author=__author__,
version=__version__,
install_requires=['pyyaml'],
test_suite=package_name + '.tests',
packages=find_packages(),
package_dir={package_name: package_name},
data_files=[
(_data_install_dir(), list(imap(_data_join, listdir(_data_join())))),
(ipynb_checkpoints_install_dir(), list(imap(ipynb_checkpoints_join, listdir(ipynb_checkpoints_join())))),
(deep_model_install_dir(), list(imap(deep_model_join, listdir(deep_model_join())))),
# (mat_scr_install_dir(), list(imap(mat_scr_join, listdir(mat_scr_join())))),
(REFUGE_result_install_dir(), list(imap(REFUGE_result_join, listdir(REFUGE_result_join())))),
(result_install_dir(), list(imap(result_join, listdir(result_join())))),
(test_img_install_dir(), list(imap(test_img_join, listdir(test_img_join()))))
]
)