Merge pull request #34 from carnal0wnage/global_variables

global vars for db_name and target

create_dbs.py

@@ -7,7 +7,6 @@ from  libs.sql import *
 #create some tables to stick data in
 
 if __name__ == "__main__":
-    db_name = "weirdAAL.db"
     create_awskey_table(db_name, "AWSKey")
     create_recon_table(db_name, "recon")
     create_services_table(db_name,"services")

libs/brute.py

@@ -17,10 +17,6 @@ import sys
 
 
 from libs.sql import *
-#from conf.conf import *
-
-# we chould probably load this from one place in the future #TODO
-db_name = "weirdAAL.db"
 
 pp = pprint.PrettyPrinter(indent=5, width=80)
 

libs/ec2.py

@@ -11,9 +11,6 @@ import time
 
 from libs.sql import *
 
-#  we chould probably load this from one place in the future #TODO
-db_name = "weirdAAL.db"
-
 pp = pprint.PrettyPrinter(indent=5, width=80)
 
 # from http://docs.aws.amazon.com/general/latest/gr/rande.html

modules/db.py

@@ -12,7 +12,6 @@ session = boto3.Session()
 credentials = session.get_credentials()
 AWS_ACCESS_KEY_ID = credentials.access_key
 
-db_name = "weirdAAL.db"
 
 #  for a key, what services does it have listed in the DB
 

test_insert.py

@@ -8,7 +8,6 @@ from  libs.sql import *
 #create some tables to stick data in
 
 if __name__ == "__main__":
-    db_name = "weirdAAL.db"
     timenow = datetime.datetime.now()
 
     test_aws_key = [("AKIAIOSFODNN7EXAMPLE", "some test shit")]

weirdAAL.py

@@ -12,6 +12,7 @@ import os
 from botocore.exceptions import ClientError
 from modules import *
 import sys
+import builtins
 
 os.environ['AWS_SHARED_CREDENTIALS_FILE'] = '.env'
 
@@ -28,6 +29,11 @@ parser.add_argument("-l", "--list", help="list modules", action="store_true")
 parser.add_argument("-v", "--verbosity", help="increase output verbosity", action="store_true")
 args = parser.parse_args()
 
+# Provides us with a global var "db_name" we can access anywhere
+builtins.db_name = "weirdAAL.py"
+
+# Provides us with a global var "target" we can access anywhere
+builtins.target = args.target
 
 def perform_credential_check():
     '''