setup.py

import os
from distutils.core import setup

import setuptools

from pygchem import __version__, __project_url__

NAME = 'PyGChem'
LIBNAME = 'pygchem'
DATA_DIR = 'data'  # path relative to LIBNAME
EXCLUDE_DIRS = ('temp',)  # paths relative to LIBNAME


def get_package_data(name, extlist):
    """Return data files for package *name* with extensions in *extlist*"""
    flist = []
    # Workaround to replace os.path.relpath (not available until Python 2.6):
    offset = len(name) + len(os.pathsep)
    for dirpath, _dirnames, filenames in os.walk(name):
        for fname in filenames:
            #if (not fname.startswith('.')
            #    and '.*' not in extlist
            #    and os.path.splitext(fname)[1] in extlist):
            #    flist.append(os.path.join(dirpath, fname)[offset:])
            flist.append(os.path.join(dirpath, fname)[offset:])
    print "Data files to install: %s" % ", ".join(flist)
    return flist


def get_subpackages(name):
    """Return subpackages of package *name*"""
    splist = []
    for dirpath, _dirnames, _filenames in os.walk(name):
        for d in EXCLUDE_DIRS:
            if dirpath == os.path.join(LIBNAME, d):
                while _dirnames: _dirnames.pop()
                continue
        if os.path.isfile(os.path.join(dirpath, '__init__.py')):
            splist.append(".".join(dirpath.split(os.sep)))
    print "Packages and subpackages to install: %s" % ", ".join(splist)
    return splist


def get_packages():
    """Return package list"""
    packages = get_subpackages(LIBNAME)  # + get_subpackages('pygchemplugins')
    return packages


setup(
    name=NAME,
    version=__version__,
    description='Python interface to GEOS-Chem 3D Chemistry Transport Model',
    long_description=open('README.md').read(),
    keywords='Interface GEOS-Chem Chemistry Transport Model CTM Atmospheric',
    author='Benoit Bovy',
    author_email='bbovy@ulg.ac.be',
    license='GPLv3',
    url=__project_url__,
    download_url='%s/archive/%s.tar.gz' % (__project_url__, __version__),
    platforms=['any'],
    include_package_data=True,
    packages=get_packages(),
    package_dir={LIBNAME: LIBNAME},
    package_data={LIBNAME: [os.path.join(DATA_DIR, '*.dat')]},
    #package_data={LIBNAME: get_package_data(os.path.join(LIBNAME, DATA_DIR),
    #                                        ('.*',))},
    #data_files=[('', ['LIBNAME/data/*.dat'])],
    scripts=[],
    requires=["numpy (>= 1.7.0)"]
)