First Commit

utils/database.py

@@ -0,0 +1,25 @@
+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