diff --git a/eb_prune/__init__.py b/eb_prune/__init__.py index 061a4d8c6399f9630498b174969a80d231c28239..ce545b3c97d34d94f098bd8f93c283ff9c273936 100644 --- a/eb_prune/__init__.py +++ b/eb_prune/__init__.py @@ -7,7 +7,12 @@ from __future__ import ( unicode_literals, ) from argparse import ArgumentParser -from botocore import session # pylint: disable=import-error +import sys + +try: + from botocore import session # pylint: disable=import-error +except ImportError: + pass def prune(versions_to_keep, dry_run): @@ -57,7 +62,7 @@ def prune(versions_to_keep, dry_run): def main(): - parser = ArgumentParser() + parser = ArgumentParser(description=__doc__) parser.add_argument( "versions_to_keep", help="The number of versions to keep.", type=int ) @@ -72,4 +77,10 @@ def main(): if __name__ == "__main__": + if "session" not in dir(): + print( + "Failed to import botocore, is the package installed?", + file=sys.stderr, + ) + sys.exit(2) main() diff --git a/setup.py b/setup.py index c2d5fdf561739129052c475fdeeafff579c1ed35..f7e0feface1d6383720989f3c640b228099a4643 100644 --- a/setup.py +++ b/setup.py @@ -1,10 +1,11 @@ #!/usr/bin/env python from setuptools import setup, find_packages +from eb_prune import __doc__ as description setup( name="eb-prune", version=open("VERSION", "r").read(), - description="Pruning of Elastic Beanstalk versions.", + description=description, long_description=open("README.rst", "r").read(), url="https://www.shore.co.il/git/eb-prune", author="Nimrod Adar",