diff --git a/modules/db.py b/modules/db.py index 536f2f9..348c759 100644 --- a/modules/db.py +++ b/modules/db.py @@ -1,3 +1,8 @@ +''' +queries that interact with db can go here +''' + + import sqlite3 from sqlite3 import Error @@ -5,8 +10,7 @@ from libs.sql import * from config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY - -#if __name__ == "__main__": +# for a key, what services does it have listed in the DB def step_show_services_by_key(): db_name = "weirdAAL.db" results = search_recon_by_key(db_name,AWS_ACCESS_KEY_ID) diff --git a/modules/sts.py b/modules/sts.py new file mode 100644 index 0000000..67a5ed3 --- /dev/null +++ b/modules/sts.py @@ -0,0 +1,12 @@ +''' +This file is used to perform some EMR actions +''' +from libs.sts import * +from config import AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY + + +def step_sts_get_accountid(): + get_accountid(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY) + +def step_sts_get_accountidall(): + get_accountid_all(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY) \ No newline at end of file