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