forked from trec-kba/kba-scorer
/
setup.py
129 lines (104 loc) · 3.27 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
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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#!/usr/bin/env python
import os
import sys
import fnmatch
import subprocess
## prepare to run PyTest as a command
from distutils.core import Command
## explain this...
#from distribute_setup import use_setuptools
#use_setuptools()
from setuptools import setup, find_packages
from version import get_git_version
VERSION = get_git_version()
PROJECT = 'kba.scorer'
AUTHOR = 'Diffeo, Inc.'
AUTHOR_EMAIL = 'trec-kba@googlegroups.com'
DESC = 'Tools for scoring output of systems built to compete in TREC KBA.'
def read_file(file_name):
file_path = os.path.join(
os.path.dirname(__file__),
file_name
)
return open(file_path).read()
def recursive_glob(treeroot, pattern):
results = []
for base, dirs, files in os.walk(treeroot):
goodfiles = fnmatch.filter(files, pattern)
results.extend(os.path.join(base, f) for f in goodfiles)
return results
def recursive_glob_with_tree(treeroot, pattern):
results = []
for base, dirs, files in os.walk(treeroot):
goodfiles = fnmatch.filter(files, pattern)
one_dir_results = []
for f in goodfiles:
one_dir_results.append(os.path.join(base, f))
results.append((base, one_dir_results))
return results
class InstallTestDependencies(Command):
'''install test dependencies'''
description = 'installs all dependencies required to run all tests'
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def easy_install(self, packages):
cmd = ['easy_install']
if packages:
cmd.extend(packages)
errno = subprocess.call(cmd)
if errno:
raise SystemExit(errno)
def run(self):
if self.distribution.install_requires:
self.easy_install(self.distribution.install_requires)
if self.distribution.tests_require:
self.easy_install(self.distribution.tests_require)
class PyTest(Command):
'''run py.test'''
description = 'runs py.test to execute all tests'
user_options = []
def initialize_options(self):
pass
def finalize_options(self):
pass
def run(self):
if self.distribution.install_requires:
self.distribution.fetch_build_eggs(
self.distribution.install_requires)
if self.distribution.tests_require:
self.distribution.fetch_build_eggs(
self.distribution.tests_require)
errno = subprocess.call([sys.executable, 'runtests.py'])
raise SystemExit(errno)
setup(
name=PROJECT,
version=VERSION,
description=DESC,
#long_description=read_file('README.md'),
long_description="",
author=AUTHOR,
author_email=AUTHOR_EMAIL,
url='',
packages = find_packages('src'),
package_dir = {'': 'src'},
cmdclass={'test': PyTest,
'install_test': InstallTestDependencies},
# We can select proper classifiers later
classifiers=[
'Development Status :: 3 - Alpha',
'Topic :: Utilities',
'License :: MIT', ## MIT/X11 license http://opensource.org/licenses/MIT
],
tests_require=[
'pytest',
],
install_requires=[
'numpy',
'matplotlib',
#'thrift',
#'streamcorpus',
],
)