setup.py

"""Setup script for cloudtracker"""
import os
import re

from setuptools import find_packages, setup


HERE = os.path.dirname(__file__)
VERSION_RE = re.compile(r"""__version__ = ['"]([0-9.]+)['"]""")
TESTS_REQUIRE = ["coverage", "nose"]


def get_version():
    init = open(os.path.join(HERE, "cloudtracker", "__init__.py")).read()
    return VERSION_RE.search(init).group(1)


def get_description():
    return open(
        os.path.join(os.path.abspath(HERE), "README.md"), encoding="utf-8"
    ).read()


setup(
    name="cloudtracker",
    version=get_version(),
    author="Duo Security",
    description=(
        "CloudTracker helps you find over-privileged IAM users and "
        "roles by comparing CloudTrail logs with current IAM policies"
    ),
    long_description=get_description(),
    long_description_content_type="text/markdown",
    url="https://github.com/duo-labs/cloudtracker",
    entry_points={"console_scripts": "cloudtracker=cloudtracker.cli:main"},
    test_suite="tests/unit",
    tests_require=TESTS_REQUIRE,
    extras_require={
        "dev": TESTS_REQUIRE + ["autoflake", "autopep8", "pylint", "invoke"],
        "es1": ["elasticsearch==1.9.0", "elasticsearch_dsl==0.0.11"],
        "es6": ["elasticsearch==6.1.1", "elasticsearch_dsl==6.1.0"],
    },
    install_requires=[
        "ansicolors==1.1.8",
        "boto3==1.5.32",
        "jmespath==0.9.3",
        "pyyaml==4.2b4",
    ],
    setup_requires=["nose"],
    packages=find_packages(exclude=["tests*"]),
    package_data={"cloudtracker": ["data/*.txt"]},
    zip_safe=True,
    license="BSD 3",
    keywords="aws cloudtracker cloudtrail privileged iam user roles policy policies",
    classifiers=[
        "License :: OSI Approved :: BSD License",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.3",
        "Programming Language :: Python :: 3.4",
        "Programming Language :: Python :: 3.5",
        "Programming Language :: Python :: 3.6",
        "Programming Language :: Python :: 3.7",
        "Programming Language :: Python :: 3 :: Only",
        "Development Status :: 5 - Production/Stable",
    ],
)