You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
208 lines
8.0 KiB
Python
208 lines
8.0 KiB
Python
"""
|
|
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 ='pystatsmodels@googlegroups.com'
|
|
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)
|