setup.py

"""A setuptools based setup module.

See:
https://packaging.python.org/en/latest/distributing.html
"""

# Always prefer setuptools over distutils
from setuptools import setup, find_packages
from setuptools.command.install import install
# To use a consistent encoding
from codecs import open
from os import path
import sys
import subprocess

here = path.abspath(path.dirname(__file__))

# Get the long description from the README file
with open(path.join(here, 'README.rst'), encoding='utf-8') as f:
    long_description = f.read()


# http://stackoverflow.com/a/287944
class bcolors:
    def __init__(self):
        pass

    HEADER = '\033[95m'
    OKBLUE = '\033[94m'
    OKGREEN = '\033[92m'
    WARNING = '\033[93m'
    FAIL = '\033[91m'
    ENDC = '\033[0m'
    BOLD = '\033[1m'
    UNDERLINE = '\033[4m'


# http://stackoverflow.com/a/4417735
def execute(cmd):
    popen = subprocess.Popen(cmd, stdout=subprocess.PIPE, universal_newlines=True)
    for stdout_line in iter(popen.stdout.readline, ""):
        yield stdout_line

    popen.stdout.close()
    return_code = popen.wait()
    if return_code:
        raise subprocess.CalledProcessError(return_code, cmd)


class CustomInstall(install):
    def run(self):
        try:
            for execute_path in execute(['sudo', 'sh', 'depend_install.sh']):
                sys.stdout.write(execute_path)

        except subprocess.CalledProcessError as e:
            print bcolors.FAIL + 'Failed to install the dependencies. Please try again.' + bcolors.ENDC
            print e
            sys.exit(1)

        try:
            for execute_path in execute(['sudo', 'sh', 'astromatic_install.sh']):
                sys.stdout.write(execute_path)

        except subprocess.CalledProcessError as e:
            print bcolors.FAIL + 'Failed to install AstrOmatic software. Please try again.' + bcolors.ENDC
            print e
            sys.exit(1)

        install.run(self)
        
        try:
            for execute_path in execute(['sudo', 'sh', 'post_install.sh']):
                sys.stdout.write(execute_path)

        except subprocess.CalledProcessError as e:
            print bcolors.FAIL + 'Failed to complete the post installation script. Please try again.' + bcolors.ENDC
            print e
            sys.exit(1)


setup(
    name='photopipe',

    # Versions should comply with PEP440.  For a discussion on single-sourcing
    # the version across setup.py and the project code, see
    # https://packaging.python.org/en/latest/single_source_version.html
    version='0.1.0b4',

    description='PhotoPipe is a pipeline for automated reduction, photometry and astrometry of imaging data ' +\
                'from RATIR and RIMAS.',
    long_description=long_description,

    # The project's main homepage.
    url='https://github.com/maxperry/photometrypipeline',

    # Author details
    author='maxperry, vickitoy, cenko',
    author_email='photopipe@googlegroups.com',

    # Choose your license
    license='GPL-3.0',

    # See https://pypi.python.org/pypi?%3Aaction=list_classifiers
    classifiers=[
        # How mature is this project? Common values are
        #   3 - Alpha
        #   4 - Beta
        #   5 - Production/Stable
        'Development Status :: 4 - Beta',

        # Indicate who your project is intended for
        'Intended Audience :: Science/Research',
        'Topic :: Scientific/Engineering :: Astronomy',

        # Pick your license as you wish (should match "license" above)
        'License :: OSI Approved :: GNU General Public License v3 (GPLv3)',

        # Specify the Python versions you support here. In particular, ensure
        # that you indicate whether you support Python 2, Python 3 or both.
        'Programming Language :: Python :: 2',
        # 'Programming Language :: Python :: 2.6',
        'Programming Language :: Python :: 2.7',
    ],

    # What does your project relate to?
    keywords='cosmology photometry reduction astrometry ratir rimas astropy',

    # You can just specify the packages manually here if your project is
    # simple. Or you can use find_packages().
    packages=find_packages(exclude=['contrib', 'docs', 'tests']),

    # Alternatively, if you want to distribute just a my_module.py, uncomment
    # this:
    #   py_modules=["my_module"],

    # List run-time dependencies here.  These will be installed by pip when
    # your project is installed. For an analysis of "install_requires" vs pip's
    # requirements files see:
    # https://packaging.python.org/en/latest/requirements.html
    # install_requires=['numpy', 'scipy', 'matplotlib', 'pyfits'],

    # List additional groups of dependencies here (e.g. development
    # dependencies). You can install these using the following syntax,
    # for example:
    # $ pip install -e .[dev,test]
    # extras_require={
    #     'dev': ['check-manifest'],
    #     'test': ['coverage'],
    # },

    # If there are data files included in your packages that need to be
    # installed, specify them here.  If using Python 2.6 or less, then these
    # have to be included in MANIFEST.in as well.
    package_data={
        'photopipe': [
            'VERSION', 
            'schemas/*.json', 
            'photometry/defaults/*.*', 
            'photometry/dependencies/*.*',
            'reduction/astrom/*.*',
            'reduction/auto/defaults/*.*'
            ],
        },

    # Although 'package_data' is the preferred approach, in some case you may
    # need to place data files outside of your packages. See:
    # http://docs.python.org/3.4/distutils/setupscript.html#installing-additional-files # noqa
    # In this case, 'data_file' will be installed into '<sys.prefix>/my_data'
    # data_files=[('my_data', ['data/data_file'])],

    # To provide executable scripts, use entry points in preference to the
    # "scripts" keyword. Entry points provide cross-platform support and allow
    # pip to create the appropriate form of executable for the target platform.
    entry_points={
        'console_scripts': [
            'photopipe=photopipe.cmdline:execute',
        ],
    },

    # http://stackoverflow.com/a/37283972
    cmdclass={
        'install': CustomInstall
    }, install_requires=['numpy', 'astropy', 'matplotlib', 'scipy']
)