setup.py

# -*- coding: utf-8 -*-
import re
import sys
from setuptools import setup
from setuptools.command.test import test as TestCommand


REQUIRES = [
    'docopt',
    'PyYAML',
    'docker',
    'termcolor',
]

class PyTest(TestCommand):
    def finalize_options(self):
        TestCommand.finalize_options(self)
        self.test_args = []
        self.test_suite = True

    def run_tests(self):
        import pytest
        errcode = pytest.main(self.test_args)
        sys.exit(errcode)


def find_version(fname):
    '''Attempts to find the version number in the file names fname.
    Raises RuntimeError if not found.
    '''
    version = ''
    with open(fname, 'r') as fp:
        reg = re.compile(r'__version__ = [\'"]([^\'"]*)[\'"]')
        for line in fp:
            m = reg.match(line)
            if m:
                version = m.group(1)
                break
    if not version:
        raise RuntimeError('Cannot find version information')
    return version

__version__ = find_version("yans.py")


def read(fname):
    with open(fname) as fp:
        content = fp.read()
    return content

setup(
    name='YANS',
    version="0.2.3",
    description='Yet Another Network Simulator',
    long_description=read("README.rst"),
    author='Kenneth Jiang',
    author_email='kenneth.jiang@gmail.com',
    url='https://github.com/kennethjiang/YANS',
    install_requires=REQUIRES,
    license=read("LICENSE"),
    zip_safe=False,
    keywords='network simulator',
    classifiers=[
        'Development Status :: 2 - Pre-Alpha',
        'Intended Audience :: Developers',
        'License :: OSI Approved :: MIT License',
        'Natural Language :: English',
        "Programming Language :: Python :: 2",
        'Programming Language :: Python :: 2.7',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.3',
        'Programming Language :: Python :: Implementation :: CPython',
        'Programming Language :: Python :: Implementation :: PyPy'
    ],
    py_modules=["yans", "docker_command", "topology"],
    entry_points={
        'console_scripts': [
            "yans = yans:main"
        ]
    },
    tests_require=['pytest'],
    cmdclass={'test': PyTest}
)