still futzing around

This commit is contained in:
cktricky
2018-09-14 11:52:27 -04:00
parent b00ee271ff
commit cec955c3c8
2 changed files with 14 additions and 7 deletions

View File

@@ -5,8 +5,13 @@ def list_all_files(directory):
array = [] array = []
for file in list_of_files: path ="modules"
filename_and_ext = os.path.splitext(file) for (dirpath, dirnames, filenames) in os.walk(path):
if (filename_and_ext[1] == ".py") and not (filename_and_ext[0].startswith("__")): if ( not (dirpath == os.path.basename(directory)) and
array.append(filename_and_ext[0]) (os.path.isdir(dirpath))
and not (os.path.basename(dirpath).startswith('__')) ):
if filenames:
list_path_name = dirpath.split('/')
array.append(".".join(list_path_name))
#array.append(directory.split)
return array return array

View File

@@ -19,10 +19,11 @@ os.environ['AWS_SHARED_CREDENTIALS_FILE'] = '.env'
# If you want to use a transparent + supports SSL proxy you can put it here # If you want to use a transparent + supports SSL proxy you can put it here
# os.environ['HTTPS_PROXY'] = 'https://127.0.0.1:3128' # os.environ['HTTPS_PROXY'] = 'https://127.0.0.1:3128'
sys.path.append("modules/aws/") sys.path.append("modules")
for module in all_modules: #for module in all_modules:
exec("from %s import *" % module) # exec("from %s import *" % module)
from modules import *
parser = argparse.ArgumentParser() 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)
@@ -56,6 +57,7 @@ def perform_credential_check():
def method_create(): def method_create():
try: try:
print(globals())
arg = globals()["module_" + args.module] arg = globals()["module_" + args.module]
return arg return arg
except KeyError: except KeyError: