-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.py
59 lines (46 loc) · 1.98 KB
/
db.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
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 item (id INTEGER PRIMARY KEY, name text, description text, "
"price text) ")
self.cur.execute(
"CREATE TABLE IF NOT EXISTS user (userid INTEGER PRIMARY KEY, name text, email text, "
"phone_num text) ")
self.conn.commit()
def fetch(self):
self.cur.execute("SELECT * FROM item")
rows = self.cur.fetchall()
return rows
def fetch2(self):
self.cur.execute("SELECT * FROM user")
rows2 = self.cur.fetchall()
return rows2
def insert(self, name, description, price):
self.cur.execute("INSERT INTO item VALUES(NULL,?,?,?)", (name, description, price))
self.conn.commit()
def insert_user(self, name, email, phone_num):
self.cur.execute("INSERT INTO user VALUES(NULL,?,?,?)", (name, email, phone_num))
self.conn.commit()
def remove(self, id):
self.cur.execute("DELETE FROM item WHERE id=?", (id,))
self.conn.commit()
def remove_user(self, userid):
self.cur.execute("DELETE FROM user WHERE userid=?", (userid,))
self.conn.commit()
def update(self, id, name, description, price):
self.cur.execute("UPDATE item SET name=?, description=?,price=? WHERE id = ?",
(name, description, price, id))
self.conn.commit()
def update_user(self, userid, name, email, phone_num):
self.cur.execute("UPDATE user SET name=?, email=?, phone_num=? WHERE userid = ?",
(name, email, phone_num, userid))
self.conn.commit()
def __del__(self):
self.conn.close()
# db = Database('thriftstore.db')
# db.insert("Shirt", "size small blue shirt", "5")
# db.insert("Pants", "size 14 straight cut jeans", "10")
# db.insert("T-shirt", "large band t", "7")