setup.py

#!/usr/bin/env python
from __future__ import print_function
import os.path
import distutils.core
import distutils.command.build

from distutils.command.build import build
import os.path
import sys

DEPENDENCIES = [
    "numpy",
    "scikit-learn==0.20",
    "scipy",
    "rolling_measures",
    "six>=1.14"
]


class BuildModelsCommand(distutils.core.Command):
    description = "Train models and save the model parameters to file"
    user_options = []
    def initialize_options(self):
        pass
    def finalize_options(self):
        pass
    def run(self):
        if os.path.exists(os.path.join(os.path.dirname(__file__), 'vessel_scoring', 'models')):
            return
        print("Training models...", end='')
        sys.stdout.flush()
        import vessel_scoring.models
        vessel_scoring.models.train_models()
        print("done.")
        sys.stdout.flush()

class build(distutils.command.build.build):
    def run(self):
        self.run_command("build_models")
        distutils.command.build.build.run(self)

cmdclass = {}
cmdclass['build_models'] = BuildModelsCommand
cmdclass['build'] = build


distutils.core.setup(
    name='vessel-scoring',
    description="Tools to score fishing behviour of vessels",
    long_description='',
    packages=[
        'vessel_scoring',
    ],
    package_data={
        'vessel_scoring': ['models/*']},
    install_requires=DEPENDENCIES,
    extras_require={
        'dev': ['matplotlib', 'ipython', 'coveralls']},
    version='2.0.0',
    author='Egil Moeller, Timothy Hochberg',
    author_email='egil@skytruth.org, tim@skytruth.org',
    url='',
    license='Apache',
    cmdclass=cmdclass
)