setup.py

#!/usr/bin/env python
import os
import sys
import numpy as np

from setuptools import setup
from setuptools.command.test import test as TestCommand

from setuptools import Extension
from Cython.Build import cythonize

try:
    from brainstorm import __about__
    about = __about__.__dict__
except ImportError:
    # installing - dependencies are not there yet
    ext_modules = []
    # Manually extract the __about__
    about = dict()
    exec(open("brainstorm/__about__.py").read(), about)


if sys.argv[-1] == 'publish':
    os.system('python setup.py sdist upload')
    sys.exit()


class PyTest(TestCommand):
    user_options = [('pytest-args=', 'a', "Arguments to pass to py.test")]

    def initialize_options(self):
        TestCommand.initialize_options(self)
        self.pytest_args = []

    def finalize_options(self):
        TestCommand.finalize_options(self)
        self.test_args = []
        self.test_suite = True

    def run_tests(self):
        # import here, cause outside the eggs aren't loaded
        import pytest
        errno = pytest.main(self.pytest_args)
        sys.exit(errno)


doclink = """
Documentation
-------------

The full documentation is at http://brainstorm.rtfd.org."""
history = open('HISTORY.rst').read().replace('.. :changelog:', '')

setup(
    name='brainstorm',
    version=about['__version__'],
    description='Fast, flexible and fun neural networks.',
    long_description=doclink + '\n\n' + history,
    author=about['__author__'],
    url=about['__url__'],
    packages=['brainstorm',
              'brainstorm.structure',
              'brainstorm.layers',
              'brainstorm.training',
              'brainstorm.handlers'],
    install_requires=['six', 'numpy', 'h5py'],
    tests_requires=['pytest', 'mock'],
    cmdclass={'test': PyTest},
    license=about['__license__'],
    classifiers=[
        'Development Status :: 4 - Beta',
        'Intended Audience :: Developers',
        'License :: OSI Approved :: MIT License',
        'Natural Language :: English',
        'Programming Language :: Python :: 2.7',
        'Programming Language :: Python :: 3.3',
    ],

    ext_modules=cythonize([Extension("brainstorm.handlers._cpuop",
                                     ["brainstorm/handlers/_cpuop.pyx"],
                                     include_dirs=[np.get_include()])]),
)