#!/usr/bin/env python

from setuptools import setup, Command

try: # Python 3.x

from distutils.command.build_py import build_py_2to3 as build_py

except ImportError: # Python 2.x

from distutils.command.build_py import build_py

class ATpyTest(Command):

user_options = []

def initialize_options(self):

pass

def finalize_options(self):

pass

def run(self):

import os

import shutil

import tempfile

# First ensure that we build the package so that 2to3 gets executed

self.reinitialize_command('build', inplace=False)

self.run_command('build')

build_cmd = self.get_finalized_command('build')

new_path = os.path.abspath(build_cmd.build_lib)

# Copy the build to a temporary directory for the purposes of testing

# - this avoids creating pyc and __pycache__ directories inside the

# build directory

tmp_dir = tempfile.mkdtemp(prefix='atpy-test-')

testing_path = os.path.join(tmp_dir, os.path.basename(new_path))

shutil.copytree(new_path, testing_path)

import sys

import subprocess

errno = subprocess.call([sys.executable, os.path.abspath('runtests.py')], cwd=testing_path)

raise SystemExit(errno)

setup(name='ATpy',

version='0.9.7',

description='Astronomical Tables in Python',

author='Thomas Robitaille and Eli Bressert',

author_email='[email protected], [email protected]',

license='MIT',

url='http://atpy.readthedocs.org/',

packages=['atpy', 'atpy.tests'],

provides=['atpy'],

requires=['numpy', 'astropy'],

cmdclass = {'build_py':build_py, 'test':ATpyTest},

keywords=['Scientific/Engineering'],

classifiers=[

"Development Status :: 4 - Beta",

"Programming Language :: Python",

"License :: OSI Approved :: MIT License",

],

)