setup.py

import os
import re
import glob
from setuptools import setup, find_packages


def find_version(*paths):
    fname = os.path.join(*paths)
    with open(fname) as fhandler:
        version_file = fhandler.read()
        version_match = re.search(r"^__VERSION__ = ['\"]([^'\"]*)['\"]",
                                  version_file, re.M)

    if not version_match:
        raise RuntimeError("Unable to find version string in %s" % (fname,))

    version = version_match.group(1)

    return version


def find_readme(*paths):
    with open(os.path.join(*paths)) as f:
        return f.read()


version = find_version('tfmesos', '__init__.py')
setup(
    name='tfmesos',
    version=version,
    packages=find_packages(),
    license='BSD License',
    description="Tensorflow on Mesos",
    long_description=find_readme('README.rst'),
    author="Zhongbo Tian",
    author_email="tianzhongbo@douban.com",
    download_url=(
        'https://github.com/douban/tfmesos/archive/%s.tar.gz' % version
    ),
    classifiers=[
        "Intended Audience :: Developers",
        "License :: OSI Approved :: BSD License",
        "Programming Language :: Python",
        "Programming Language :: Python :: Implementation :: CPython",
        "Operating System :: POSIX :: Linux",
        "Topic :: Software Development :: Libraries"
    ],
    install_requires=[
        'six',
        'addict',
        'pymesos>=0.2.2',
    ],
    extras_require={
        'cpu': ['tensorflow>=0.8.0'],
        'gpu': ['tensorflow-gpu>=0.8.0'],
    },
    scripts=glob.glob(os.path.join('script', '*')),
)