From 250f1b83d62a49e8dec87f54971d66e61330f2b6 Mon Sep 17 00:00:00 2001 From: conor Date: Mon, 29 Jul 2019 15:01:15 +0200 Subject: [PATCH] A little error handling --- atmos.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/atmos.py b/atmos.py index 765fc30..cc2133d 100755 --- a/atmos.py +++ b/atmos.py @@ -83,8 +83,16 @@ def generate_creds(args): print(secret_key_name) contents = contents + "[{workspace}]\n".format(workspace=workspace) - contents = contents + "aws_access_key_id=" + os.environ.get(access_key_name) + "\n" - contents = contents + "aws_secret_access_key=" + os.environ.get(secret_key_name) + "\n" + try: + contents = contents + "aws_access_key_id=" + os.environ.get(access_key_name) + "\n" + except: + print("[ERROR]: Env Variable " + access_key_name + " not found.") + sys.exit(1) + try: + contents = contents + "aws_secret_access_key=" + os.environ.get(secret_key_name) + "\n" + except: + print("[ERROR]: Env Variable " + secret_key_name + " not found.") + sys.exit(1) with open(os.path.expanduser('~/.aws/credentials-atmos'), 'w+') as f: f.write(contents)