setup.py

#!/usr/bin/env python
from __future__ import print_function

import os
import shutil
import sys
import glob

from setuptools import setup, Command, find_packages
from setuptools.command.install import install

INFO = {
    'version': '0.1.0',
}

paths = """
import os

fullpath = lambda *x: os.path.abspath(os.path.join(*x))

ROOT_DIR = fullpath(r'{here}')
MODEL_DIR = fullpath(r'{here}', 'model')
DATA_DIR = fullpath(r'{here}', 'model', 'data')
OUTPUT_DIR = fullpath(r'{here}', 'model', 'output')
MSG_TEST_DIR = fullpath(r'{here}', 'tests')

""".format(here=os.path.dirname(os.path.realpath(__file__)))


class Cmd(install):
    """Custom clean command to tidy up the project root."""

    def initialize_options(self):
        install.initialize_options(self)

    def finalize_options(self):
        install.finalize_options(self)

    def run(self):
        install.run(self)
        dirs = [
            'message_ix.egg-info'
        ]
        for d in dirs:
            print('removing {}'.format(d))
            shutil.rmtree(d)


def main():
    packages = [
        'message_ix'
    ]
    pack_dir = {
        'message_ix': 'message_ix',
    }
    entry_points = {
        'console_scripts': [
        ],
    }
    cmdclass = {
        'install': Cmd,
    }
    pack_data = {}
    setup_kwargs = {
        "name": "message_ix",
        "version": INFO['version'],
        "description": 'The MESSAGEix Integrated Assessment Model',
        "author": 'Daniel Huppmann, Matthew Gidden, Volker Krey,  '
                  'Oliver Fricko, Peter Kolp',
        "author_email": 'message_ix@iiasa.ac.at',
        "url": 'http://github.com/iiasa/message_ix',
        "packages": packages,
        "package_dir": pack_dir,
        "package_data": pack_data,
        "entry_points": entry_points,
        "cmdclass": cmdclass,
    }
    print('Writing default_paths.py')
    pth = os.path.join('message_ix', 'default_paths.py')
    with open(pth, 'w') as f:
        f.write(paths)
    rtn = setup(**setup_kwargs)
    print('removing default_paths.py')
    os.remove(pth)

if __name__ == "__main__":
    main()