stab at ec2 log sub_service to db

This commit is contained in:
carnal0wnage
2018-04-18 21:49:59 -04:00
parent 51b6a6d205
commit 04e2fb230f
4 changed files with 47 additions and 5 deletions

View File

@@ -27,6 +27,7 @@ def create_table(db_name,table_name,sql):
cursor.execute(sql)
db.commit()
def create_recon_table(db_name, table_name):
sql = """CREATE TABLE recon
(ID integer,
@@ -39,6 +40,7 @@ def create_recon_table(db_name, table_name):
create_table(db_name,table_name,sql)
print ("created table: {}".format(table_name))
def create_awskey_table(db_name, table_name):
sql = """CREATE TABLE AWSKey
(ID integer,
@@ -48,6 +50,7 @@ def create_awskey_table(db_name, table_name):
create_table(db_name,table_name,sql)
print ("created table: {}".format(table_name))
def create_services_table(db_name, table_name):
sql = """CREATE TABLE services
(ID integer,
@@ -66,11 +69,19 @@ def insert_awskey_data(db_name, records):
for record in records:
query(db_name, sql,record)
def insert_reconservice_data(db_name, records):
sql = """INSERT INTO recon(service, sub_service, AWSKeyID, checked_at) VALUES (?,?,?,?)"""
for record in records:
query(db_name,sql,record)
def insert_sub_service_data(db_name, records):
sql = """INSERT INTO services(service, sub_service, sub_service_data, AWSKeyID, checked_at) VALUES (?,?,?,?,?)"""
for record in records:
query(db_name,sql,record)
def search_recon_by_key(db_name,AWSKeyID):
with sqlite3.connect(db_name) as db:
cursor = db.cursor()
@@ -78,6 +89,7 @@ def search_recon_by_key(db_name,AWSKeyID):
results = cursor.fetchall()
return results
def query(db_name,sql,data):
with sqlite3.connect(db_name) as db:
cursor = db.cursor()