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

fixed up legacy/remaining instances of step

Browse Source
This commit is contained in:
cktricky
2018-04-19 11:56:13 -04:00
parent e7eb42eb87
commit 694c72560b
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
weirdAAL.py
View File

@@ -24,7 +24,7 @@ parser = argparse.ArgumentParser()
parser.add_argument("-m", "--module", help="list the module you would like to run", action="store", type=str, required=True) parser.add_argument("-m", "--module", help="list the module you would like to run", 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", 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("-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 modules", action="store_true")
parser.add_argument("-v", "--verbosity", help="increase output verbosity", action="store_true") parser.add_argument("-v", "--verbosity", help="increase output verbosity", action="store_true")
args = parser.parse_args() args = parser.parse_args()
@@ -47,10 +47,10 @@ def perform_credential_check():
def method_create(): def method_create():
try: try:
arg = globals()["module_" + args.step] arg = globals()["module_" + args.module]
return arg return arg
except KeyError: except KeyError:
print("That step does not exist") print("That module does not exist")
exit(1) exit(1)
@@ -71,8 +71,8 @@ arg_list = None
if (args.arguments): if (args.arguments):
arg_list = args.arguments.split(',') arg_list = args.arguments.split(',')
# We need the user to tell us the step they want to proceed on # We need the user to tell us the module they want to proceed on
if (args.step): if (args.modules):
arg = method_create() arg = method_create()
if callable(arg): if callable(arg):
if arg_list: if arg_list: