import sqlite3




class Database:
    def __init__(self, db):
        self.conn = sqlite3.connect(db)
        self.cur = self.conn.cursor()
        self.cur.execute("CREATE TABLE IF NOT EXISTS parts (id INTEGER PRIMARY KEY, part text, customer text, retailer text, price text)")
        self.conn.commit()
        
    def fetch(self):
        self.cur.execute("SELECT * FROM parts")
        rows = self.cur.fetchall()
        return rows
    
    def insert(self, part, customer, retailer, price):
        self.cur.execute("INSERT INTO parts VALUES (NULL, ?, ?, ?, ?)", (part, customer, retailer, price))
        self.conn.commit()
        
    def remove(self, id):
        self.cur.execute("DELETE FROM parts WHERE id=?", (id,))
        self.conn.commit()
        
    def update(self, id, part, customer, retailer, price):
        self.cur.execute("UPDATE parts SET part = ?, customer = ?, retailer = ?, price = ? WHERE id = ?", (part, customer, retailer, price, id))
        self.conn.commit()
        
    def __del__(self):
        self.conn.close()
        

# db = Database('store.db')
# db.insert("4GB DDR4 Ram","John Doe","Microcenter","160")
# db.insert("ASUS MOBO","Mike Henry","Microcenter","360")
# db.insert("500W PSU","Karen Johnson","NewEgg","80")
# db.insert("2GB DDR4 Ram","Karen Johnson","NewEgg","78")
# db.insert("24 Inch Samsung Monitor","Sam Smith","Best Buy","180")
# db.insert("NVIDIA RTX 2080","Albert Kingston","NewEgg","679")
# db.insert("600W Corsair PSU","Karen Johnson","NewEgg","130")