"""Create IAM Instance Profiles, Roles, Users, and Groups."""
import argparse
import logging

from .create_iam import create_iam_resources

LOG = logging.getLogger(__name__)


def main():
    """Command to create IAM Instance Profiles, Roles, Users, and Groups."""
    logging.basicConfig()

    parser = argparse.ArgumentParser(description=main.__doc__)
    parser.add_argument('-d',
                        '--debug',
                        action='store_const',
                        const=logging.DEBUG,
                        default=logging.INFO,
                        help='Set DEBUG output')
    parser.add_argument('-e',
                        '--env',
                        choices=('dev', 'stage', 'prod'),
                        default='dev',
                        help='Deploy environment')
    parser.add_argument('-a',
                        '--app',
                        default='testapp',
                        help='Spinnaker Application name')
    args = parser.parse_args()

    LOG.setLevel(args.debug)
    logging.getLogger(__package__).setLevel(args.debug)
    vars(args).pop('debug')

    assert create_iam_resources(env=args.env, app=args.app)


if __name__ == '__main__':
    main()

"""Create IAM Instance Profiles, Roles, Users, and Groups."""
import argparse
import logging

from .create_iam import create_iam_resources

LOG = logging.getLogger(__name__)


def main():
    """Command to create IAM Instance Profiles, Roles, Users, and Groups."""
    logging.basicConfig()

    parser = argparse.ArgumentParser(description=main.__doc__)
    parser.add_argument('-d',
                        '--debug',
                        action='store_const',
                        const=logging.DEBUG,
                        default=logging.INFO,
                        help='Set DEBUG output')
    parser.add_argument('-e',
                        '--env',
                        choices=('build', 'dev', 'stage', 'prod'),
                        default='dev',
                        help='Deploy environment')
    parser.add_argument('-a',
                        '--app',
                        default='testapp',
                        help='Spinnaker Application name')
    args = parser.parse_args()

    LOG.setLevel(args.debug)
    logging.getLogger(__package__).setLevel(args.debug)
    vars(args).pop('debug')

    assert create_iam_resources(env=args.env, app=args.app)


if __name__ == '__main__':
    main()

Add build as available environment