looks legit to me (trying to make sure we can load all files in the modules directory as well as libs but do it in a somewhat abstracted way. Although, the method to do the abstraction then requires duplicate code so I am sort of annoyed that I used dup code to avoid making dup code... whatever... get over it you neurotic ginger.)

This commit is contained in:
cktricky
2018-04-06 00:00:01 -10:00
parent 3b60996fdc
commit 7ff5319252
6 changed files with 36 additions and 11 deletions

View File

@@ -1,11 +1,11 @@
import os
list_of_files = os.listdir('.')
from libs.utils.common import *
arry = []
for file in list_of_files:
filename_and_ext = os.path.splitext(file)
if filename_and_ext[1] == ".py":
arry.append(filename_and_ext[0])
# Get the application's path (wherever weirdAAL.py is located will be the dirpath )
dirpath = os.getcwd()
# The actual location of this file on the filesystem is the "foldername"
foldername = os.path.dirname(os.path.realpath(__file__))
__all__ = arry
all_files = list_all_files(foldername)
__all__ = all_files

0
libs/utils/__init__.py Normal file
View File

12
libs/utils/common.py Normal file
View File

@@ -0,0 +1,12 @@
import os
def list_all_files(directory):
list_of_files = os.listdir(directory)
array = []
for file in list_of_files:
filename_and_ext = os.path.splitext(file)
if filename_and_ext[1] == ".py":
array.append(filename_and_ext[0])
return array

11
modules/__init__.py Normal file
View File

@@ -0,0 +1,11 @@
import os
from libs.utils.common import *
# Get the application's path (wherever weirdAAL.py is located will be the dirpath )
dirpath = os.getcwd()
# The actual location of this file on the filesystem is the "foldername"
foldername = os.path.dirname(os.path.realpath(__file__))
all_files = list_all_files(foldername)
__all__ = all_files

View File

@@ -5,9 +5,11 @@ from libs.iam import *
from libs.sts import *
from config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY
get_accountid(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
check_root_account(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
get_password_policy(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
def step_cg_test():
get_accountid(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
check_root_account(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
get_password_policy(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
#create_access_key(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY,'pythons3')
#delete_access_key(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY,'pythons3', 'AKIAIJV3RQMOYM7WQS2Q')
#change_user_console_password(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY, 'pythons3', 'PS#EDCasd123456!@')

View File

@@ -11,7 +11,7 @@ import argparse
import os
from config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY
from botocore.exceptions import ClientError
#from modules import *
from modules import *
parser = argparse.ArgumentParser()
parser.add_argument("-s", "--step", help="list the step you would like to run",