setup.py

from setuptools import setup, find_packages
from setuptools.command.test import test as TestCommand
import sys
import os


__version__ = '0.9.0'


class PyTest(TestCommand):
    user_options = [('pytest-args=', 'a', "Arguments to pass to py.test")]

    def initialize_options(self):
        TestCommand.initialize_options(self)
        self.pytest_args = []

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

    def run_tests(self):
        import pytest
        errno = pytest.main(self.pytest_args)
        sys.exit(errno)


def read_file(filename):
    try:
        with open(os.path.join(os.path.dirname(__file__), filename)) as f:
            return f.read()
    except IOError:
        return ''


setup(
    setup_requires=['setuptools>=17.1'],

    name='pwnypack',
    packages=find_packages(),
    version=__version__,
    description='Official Certified Edible Dinosaurs CTF toolkit.',
    long_description=read_file('README.rst') + '\n' + read_file('changelog.rst'),
    author='Ingmar Steen',
    author_email='iksteen@gmail.com',
    url='https://github.com/edibledinos/pwnypack/',
    download_url='https://github.com/edibledinos/pwnypack/tarball/v%s' % __version__,
    install_requires=['six', 'kwonly-args'],
    extras_require={
        ':python_version<"2.7"': ['counter', 'ordereddict', 'argparse'],
        ':python_version<"3.4"': ['enum34'],
        ':python_version<"3.3"': ['shutilwhich'],
        'asm': ['keystone-engine'],
        'disasm': ['capstone'],
        'pwnbook:python_version>="2.7"': ['jupyter'],
        'rop': ['capstone'],
        'shell:python_version>="2.7"': ['ipython'],
        'ssh:python_version>="2.7"': ['paramiko'],
        'all': ['capstone', 'keystone-engine'],
        'all:python_version>="2.7"': ['ipython', 'jupyter', 'paramiko'],
    },
    tests_require=['mock', 'pytest-cov', 'pytest'],
    cmdclass = {'test': PyTest},
    entry_points={
        'console_scripts': [
            'pwny=pwnypack.main:main',
        ],
    },
    keywords=['wargame', 'ctf'],
    classifiers=[
        'Environment :: Console',
        'Intended Audience :: Developers',
        'Intended Audience :: Science/Research',
        'Intended Audience :: System Administrators',
        'License :: OSI Approved :: MIT License',
        'Operating System :: OS Independent',
        'Programming Language :: Python',
        'Programming Language :: Python :: 2',
        'Programming Language :: Python :: 3',
        'Topic :: Security',
        'Topic :: Security :: Cryptography',
    ],
)