setup.py

from setuptools import setup
import distutils.cmd
import distutils.log
import os
from setuptools.command.install import install
# from pip.req import parse_requirements
from pip._internal.req import parse_requirements
import numpy as np


class InstallCommand(install):
    """A custom command to install requested package to run Survey Strategy Support Pipeline"""

    description = 'Script to install requested package to run Survey Strategy Support Pipeline'
    user_options = install.user_options + [
        ('package=', None, 'package to install')
    ]

    def initialize_options(self):
        """Set default values for options."""
        # Each user option must be listed here with their default value.
        install.initialize_options(self)
        self.package = ''
        # self.available_packages = ''
        # load available packages and versions
        self.packs = np.loadtxt('pack_version.txt', dtype={'names': (
            'packname', 'version'), 'formats': ('U15', 'U15')})

    def finalize_options(self):
        """Post-process options."""

        if self.package:
            if self.package not in self.packs['packname'].tolist() and self.package != 'sn_pipe':
                print('{} impossible to install'.format(self.package))
        install.finalize_options(self)

    def run(self):
        # install dependencies first
        if self.package == 'ztf_pipe':
            cmd = 'pip install --user -r requirements.txt --no-deps'
            os.system(cmd)

        else:
            if self.package != 'all':
                # now install the requested package
                # get the version for this package
                idx = self.packs['packname'] == self.package
                version = self.packs[idx]['version'].item()
                cmd = 'pip install -v --user git+https://github.com/lsstdesc/{}.git@{}'.format(
                    self.package, version)
                os.system(cmd)
            else:
                for pack in self.packs:
                    # get the version for this package
                    packname = pack['packname']
                    version = pack['version']
                    cmd = 'pip install --user git+https://github.com/lsstdesc/{}.git@{}'.format(
                        packname, version)
                    os.system(cmd)

        install.run(self)


# get the version here
pkg_vars = {}

with open("version.py") as fp:
    exec(fp.read(), pkg_vars)

setup(
    name='sn_pipe',
    version=pkg_vars['__version__'],
    description='A framework to run the Survey Strategy Support pipeline for supernovae',
    url='http://github.com/lsstdesc/sn_pipe',
    author='Philippe Gris',
    author_email='philippe.gris@clermont.in2p3.fr',
    license='BSD',
    python_requires='>=3.5',
    zip_safe=False,
    cmdclass={
        'install': InstallCommand,
    },

)