forked from statsmodels/statsmodels
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
207 lines (184 loc) · 7.81 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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
"""
setuptools must be installed first. If you do not have setuptools installed
please download and install it from http://pypi.python.org/pypi/setuptools
"""
import os
import sys
import subprocess
import setuptools
from numpy.distutils.core import setup
import numpy
compile_cython = 0
if "--with-cython" in sys.argv:
compile_cython = 1
sys.argv.remove('--with-cython')
curdir = os.path.abspath(os.path.dirname(__file__))
README = open(os.path.join(curdir, "README.txt")).read()
CHANGES = open(os.path.join(curdir, "CHANGES.txt")).read()
DISTNAME = 'scikits.statsmodels'
DESCRIPTION = 'Statistical computations and models for use with SciPy'
LONG_DESCRIPTION = README + '\n\n' + CHANGES
MAINTAINER = 'Skipper Seabold, Josef Perktold'
MAINTAINER_EMAIL ='[email protected]'
URL = 'http://statsmodels.sourceforge.net/'
LICENSE = 'BSD License'
DOWNLOAD_URL = ''
MAJ = 0
MIN = 4
REV = 0
ISRELEASED = False
VERSION = '%d.%d.%d' % (MAJ,MIN,REV)
classifiers = [ 'Development Status :: 4 - Beta',
'Environment :: Console',
'Programming Language :: Python :: 2.5',
'Programming Language :: Python :: 2.6',
'Programming Language :: Python :: 2.7',
'Programming Language :: Python :: 3.2',
'Operating System :: OS Independent',
'Intended Audience :: Developers',
'Intended Audience :: Science/Research',
'License :: OSI Approved :: BSD License',
'Topic :: Scientific/Engineering']
# Return the git revision as a string
def git_version():
def _minimal_ext_cmd(cmd):
# construct minimal environment
env = {}
for k in ['SYSTEMROOT', 'PATH']:
v = os.environ.get(k)
if v is not None:
env[k] = v
# LANGUAGE is used on win32
env['LANGUAGE'] = 'C'
env['LANG'] = 'C'
env['LC_ALL'] = 'C'
out = subprocess.Popen(" ".join(cmd), stdout = subprocess.PIPE, env=env,
shell=True).communicate()[0]
return out
try:
out = _minimal_ext_cmd(['git', 'rev-parse', 'HEAD'])
GIT_REVISION = out.strip().decode('ascii')
except OSError:
GIT_REVISION = "Unknown"
return GIT_REVISION
def write_version_py(filename='scikits/statsmodels/version.py'):
cnt = """
# THIS FILE IS GENERATED FROM SETUP.PY
short_version = '%(version)s'
version = '%(version)s'
full_version = '%(full_version)s'
git_revision = '%(git_revision)s'
release = %(isrelease)s
if not release:
version = full_version
"""
# Adding the git rev number needs to be done inside write_version_py(),
# otherwise the import of numpy.version messes up the build under Python 3.
FULLVERSION = VERSION
dowrite = True
if os.path.exists('.git'):
GIT_REVISION = git_version()
elif os.path.exists(filename):
# must be a source distribution, use existing version file
try:
from scikits.statsmodels.version import git_revision as GIT_REVISION
print "debug import success GIT_REVISION", GIT_REVISION
except ImportError:
dowrite = False
#changed: if we are not in a git repository then don't update version.py
## raise ImportError("Unable to import git_revision. Try removing " \
## "scikits/statsmodels/version.py and the build directory " \
## "before building.")
else:
GIT_REVISION = "Unknown"
if not ISRELEASED:
FULLVERSION += '.dev-' + GIT_REVISION[:7]
if dowrite:
a = open(filename, 'w')
try:
a.write(cnt % {'version': VERSION,
'full_version' : FULLVERSION,
'git_revision' : GIT_REVISION,
'isrelease': str(ISRELEASED)})
finally:
a.close()
def configuration(parent_package='', top_path=None, package_name=DISTNAME):
#if os.path.fexists('MANIFEST'): os.remove('MANIFEST')
from numpy.distutils.misc_util import Configuration
config = Configuration(None, parent_package, top_path,
namespace_packages = ['scikits'],
name = DISTNAME,
version = VERSION,
maintainer = MAINTAINER,
maintainer_email = MAINTAINER_EMAIL,
description = DESCRIPTION,
license = LICENSE,
url = URL,
download_url = DOWNLOAD_URL,
long_description = LONG_DESCRIPTION)
config.add_subpackage('scikits')
config.add_data_files('scikits/__init__.py')
config.add_data_dir('scikits/statsmodels/tests')
config.add_data_dir('scikits/statsmodels/examples')
config.add_data_dir('scikits/statsmodels/sandbox/examples')
config.add_data_dir('scikits/statsmodels/docs')
config.add_data_dir('scikits/statsmodels/iolib/tests')
config.add_data_dir('scikits/statsmodels/discrete/tests')
config.add_data_dir('scikits/statsmodels/genmod/tests')
config.add_data_dir('scikits/statsmodels/regression/tests')
config.add_data_dir('scikits/statsmodels/robust/tests')
config.add_data_dir('scikits/statsmodels/tsa/vector_ar/tests')
config.add_data_dir('scikits/statsmodels/tsa/filters/tests')
config.add_data_files('scikits/statsmodels/docs/build/htmlhelp/statsmodelsdoc.chm')
config.add_data_files('scikits/statsmodels/iolib/tests/results/macrodata.npy')
config.add_data_dir('scikits/statsmodels/nonparametric/tests')
vardatafiles = [os.path.join(r,d) for r,ds,f in \
os.walk('scikits/statsmodels/tsa/vector_ar/data')
for d in f if not os.path.splitext(d)[1] in ['.py',
'.pyc']]
for f in vardatafiles:
config.add_data_files(f)
extradatafiles = [os.path.join(r,d) for r,ds,f in \
os.walk('scikits/statsmodels/datasets')
for d in f if not os.path.splitext(d)[1] in
['.py', '.pyc']]
for f in extradatafiles:
config.add_data_files(f)
tsaresultsfiles = [os.path.join(r,d) for r,ds,f in \
os.walk('scikits/statsmodels/tsa/tests/results') for \
d in f if not os.path.splitext(d)[1] in ['.py',
'.do', '.pyc', '.swp']]
for f in tsaresultsfiles:
config.add_data_files(f)
kderesultsfiles = [os.path.join(r,d) for r,ds,f in \
os.walk('scikits/statsmodels/nonparametric/tests/results') for \
d in f if not os.path.splitext(d)[1] in ['.py',
'.do', '.pyc', '.swp']]
for f in kderesultsfiles:
config.add_data_files(f)
if compile_cython:
config.add_extension('tsa/kalmanf/kalman_loglike',
sources = ['scikits/statsmodels/tsa/kalmanf/kalman_loglike.c'],
include_dirs=[numpy.get_include()])
#config.add_subpackage(DISTNAME)
#config.add_subpackage('scikits/statsmodels/examples')
#config.add_subpackage('scikits/statsmodels/tests')
config.set_options(
ignore_setup_xxx_py = True,
assume_default_configuration = True,
delegate_options_to_subpackages = True,
quiet = False,
)
return config
if __name__ == "__main__":
write_version_py()
setup(configuration = configuration,
#name = DISTNAME,
install_requires = ['pandas >= 0.3.0'],
namespace_packages = ['scikits'],
packages = setuptools.find_packages(),
include_package_data = True,
test_suite="nose.collector",
zip_safe = False, # the package can not run out of an .egg file bc of
# nose tests
classifiers = classifiers)