pull in .env instead of config.py

This commit is contained in:
Chris Gates
2018-04-16 14:29:36 -04:00
parent 581dea9cbc
commit 22bd12dcab

View File

@@ -14,6 +14,8 @@ from botocore.exceptions import ClientError
from modules import * from modules import *
import sys import sys
os.environ['AWS_SHARED_CREDENTIALS_FILE'] = '.env'
sys.path.append("modules") sys.path.append("modules")
for module in all_modules: for module in all_modules:
exec("from %s import *"%module) exec("from %s import *"%module)
@@ -33,11 +35,15 @@ args = parser.parse_args()
def perform_credential_check(): def perform_credential_check():
try: try:
client = boto3.client("sts", aws_access_key_id=AWS_ACCESS_KEY_ID, aws_secret_access_key=AWS_SECRET_ACCESS_KEY) #client = boto3.client("sts", aws_access_key_id=AWS_ACCESS_KEY_ID, aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
client = boto3.client("sts")
account_id = client.get_caller_identity()["Account"] account_id = client.get_caller_identity()["Account"]
except botocore.exceptions.NoCredentialsError as e:
print("Error: Unable to locate credentials")
sys.exit("fix your credentials file -exiting...")
except ClientError as e: except ClientError as e:
print("The AWS Access Keys are not valid/active") print("The AWS Access Keys are not valid/active")
#exit(1) sys.exit(1)
def step_recon(): def step_recon():
print("!!!") print("!!!")