dash/weirdAAL
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pep8 and some comments

Browse Source
This commit is contained in:
Chris Gates
2018-04-16 16:47:52 -04:00
parent 886ba10e49
commit b928189a66
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
weirdAAL.py
View File

@@ -17,24 +17,25 @@ 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)
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("-s", "--step", help="list the step you would like to run", parser.add_argument("-s", "--step", help="list the step you would like to run", action="store", type=str, required=True)
action="store", type=str, required=True) parser.add_argument("-t", "--target", help="Give your target a name so we can track results", action="store", type=str, required=True)
parser.add_argument("-t", "--target", help="Give your target a name so we can track results", parser.add_argument("-a", "--arguments", help="Provide a list of arguments, comma separated. Ex: arg1,arg2,arg3", action="store", type=str, required=False)
action="store", type=str, required=True)
parser.add_argument("-a", "--arguments", help="Provide a list of arguments, comma separated. Ex: arg1,arg2,arg3",
action="store", type=str, required=False)
parser.add_argument("-l", "--list", help="list steps", action="store_true") parser.add_argument("-l", "--list", help="list steps", action="store_true")
parser.add_argument("-v", "--verbosity", help="increase output verbosity", parser.add_argument("-v", "--verbosity", help="increase output verbosity", action="store_true")
action="store_true")
args = parser.parse_args() args = parser.parse_args()
def perform_credential_check(): def perform_credential_check():
'''
Check that the AWS keys work before we go any further. It picks the keys up from the local .env file
We are letting boto3 do all the work that way we can handle session tokens natively
'''
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") client = boto3.client("sts")
account_id = client.get_caller_identity()["Account"] account_id = client.get_caller_identity()["Account"]
except botocore.exceptions.NoCredentialsError as e: except botocore.exceptions.NoCredentialsError as e:
@@ -44,9 +45,11 @@ def perform_credential_check():
print("The AWS Access Keys are not valid/active") print("The AWS Access Keys are not valid/active")
sys.exit(1) sys.exit(1)
def step_recon(): def step_recon():
print("!!!") print("!!!")
def method_create(): def method_create():
try: try:
arg = globals()["step_" + args.step] arg = globals()["step_" + args.step]
@@ -55,12 +58,13 @@ def method_create():
print("That step does not exist") print("That step does not exist")
exit(1) exit(1)
# Need to figure out if we have keys in the ENV or not # Need to figure out if we have keys in the ENV or not
if AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY: if AWS_ACCESS_KEY_ID and AWS_SECRET_ACCESS_KEY:
perform_credential_check() perform_credential_check()
else: else:
print("Please supply keys as outlined in our README.md file") print("Please supply keys as outlined in our README.md file")
#exit(1) # exit(1)
if (args.list): if (args.list):
pass pass
@@ -79,7 +83,7 @@ if (args.step):
if arg_list: if arg_list:
arg(arg_list) arg(arg_list)
else: else:
arg() arg()
# Allow the user to specify verbosity for debugging # Allow the user to specify verbosity for debugging