brotoskyj
25 lines
873 B
Python
25 lines
873 B
Python
import pathlib
|
|
import sqlite3
|
|
|
|
conn = sqlite3.connect('database.db')
|
|
cursor = conn.cursor()
|
|
|
|
def createdatabasefile():
|
|
pathlib.Path("database.db").touch(exist_ok=True)
|
|
cursor.execute("SELECT * FROM sqlite_master WHERE type='table' AND name='active_attacks'")
|
|
result = cursor.fetchone()
|
|
if result is None:
|
|
cursor.execute('''CREATE TABLE active_attacks
|
|
(vendor_name text, date_accessed datetime)
|
|
''')
|
|
conn.commit()
|
|
|
|
def checkdatabase(victimname, date_discovered):
|
|
query = f"SELECT vendor_name FROM active_attacks WHERE vendor_name= ?"
|
|
cursor.execute(query, (victimname,))
|
|
result = cursor.fetchone()
|
|
if result is None:
|
|
query = f"INSERT INTO active_attacks VALUES ('{victimname}', '{date_discovered}')"
|
|
cursor.execute(query)
|
|
conn.commit()
|
|
return False |