import sys
import boto3

if len(sys.argv) < 2:
	raise Exception("missing aws account as input arg")

account = sys.argv[1] 
client = boto3.client('sts')
response = client.get_caller_identity()
curr_account = response['Account']
if curr_account != account:
	raise Exception("AWS credentials incorrect")
