-
Notifications
You must be signed in to change notification settings - Fork 0
/
menuManager.py
72 lines (52 loc) · 1.48 KB
/
menuManager.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
import SQLExceptions
import unittest
def getMenu():
# fetch 'em all
conn = sqlite3.connect('exemple.db')
c = conn.cursor()
rows = c.execute ("SELECT * FROM Menu")
return_string = "{"
for row in rows:
return_string = return_string + str(row) + ","
return_string = return_string[:-1] + '}'
if '{' not in return_string:
return_string = '{' + return_string
conn.commit()
conn.close()
return return_string
def insertDish(dish_name, description, gluten, vegan, vegetarian, lactose,
type_, discount,price):
gluten = int(bool(gluten))
vegan = int(bool(vegan))
vegetarian = int(bool(vegetarian))
lactose = int(bool(lactose))
print discount
discount= float(discount)
print price
price = float(price)
conn = sqlite3.connect('exemple.db')
c = conn.cursor()
row = c.execute("""INSERT OR REPLACE INTO menu VALUES('%s','%s',%d,%d,%d,%d,'%s',%f,%f)"""
%(dish_name, description, gluten, vegan, vegetarian, lactose,
type_, discount,price))
conn.commit()
conn.close()
pass
def deleteDish(dish_name):
conn = sqlite3.connect('exemple.db')
c = conn.cursor()
found = False
for a_id in c.execute("""SELECT dish_name FROM menu
WHERE '%s'==dish_name"""%(dish_name)):
if dish_name in a_id:
found = True
if found == True:
row = c.execute("""DELETE FROM menu WHERE '%s'==dish_name"""
%(dish_name))
conn.commit()
else:
conn.close()
raise SQLExceptions.SQLTableDeletionError()
conn.close()
pass