stab at ec2 log sub_service to db
This commit is contained in:
12
libs/sql.py
12
libs/sql.py
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user