From df3704e6314406228066147e7529ec794187d7e3 Mon Sep 17 00:00:00 2001 From: PeckTonium <30939668+PeckTonium@users.noreply.github.com> Date: Wed, 23 Aug 2017 14:11:45 -0700 Subject: [PATCH] Update nextbuildver.py --- engagement/email/nextbuildver.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/engagement/email/nextbuildver.py b/engagement/email/nextbuildver.py index 6322d75..337ea82 100644 --- a/engagement/email/nextbuildver.py +++ b/engagement/email/nextbuildver.py @@ -31,6 +31,7 @@ import sys import boto3 +import botocore import semver parser = argparse.ArgumentParser(description='Find next version number') @@ -48,11 +49,14 @@ ###### Check parameter values -s3 = boto3.resource('s3') -buckets = [bucket.name for bucket in list(s3.buckets.all())] -if args.bucket_name not in buckets: +s3_client = boto3.client('s3') +try: + s3_client.head_bucket(Bucket=args.bucket_name) +except botocore.exceptions.ClientError as clerr: print "Error: \"bucket_name\" has an invalid value (%s)" % args.bucket_name exit(1) + +s3 = boto3.resource('s3') ext_index = args.artifact_name.rfind(".") if ext_index != -1: