· 5 years ago · Apr 22, 2020, 01:52 PM
1import sqlite3 as sq
2import os
3
4def create_database():
5 conn = sq.connect('test.db')
6 # print("Created Database")
7 return conn
8
9def create_table(conn):
10 sql = '''CREATE TABLE IF NOT EXISTS food_intake (
11 id INTEGER PRIMARY KEY AUTOINCREMENT,
12 food_name TEXT NOT NULL,
13 amount TEXT NOT NULL,
14 food_had DEFAULT CURRENT_TIMESTAMP
15 )'''
16 conn.execute(sql)
17
18class DietCounter:
19 def __init__(self,conn):
20 self.conn = conn
21
22 def addFood(self,food_name,amount):
23 # x.addFood('Breads,eggs,(tomato and onion)','3/1/(1/2)')
24 sql = "INSERT INTO food_intake (food_name,amount) VALUES ('{}','{}')".format(food_name,amount)
25 self.conn.execute(sql)
26 self.conn.commit()
27
28 def listAllFood(self,date_='today'):
29 if date_ == 'today':
30 sql = "SELECT * FROM food_intake WHERE food_had >= DATE('now')"
31 else:
32 sql = "SELECT * FROM food_intake"
33 cursor = self.conn.execute(sql)
34 print("\n=============================\n")
35 for row in cursor:
36 foods = row[1].split(',')
37 quantity = row[2].split(',')
38 i,s = 0,[]
39 while i < len(foods):
40 s.append(str(foods[i])+'-->'+str(quantity[i]))
41 i+=1
42 print("FOODS: ",s,"TIME: ",row[-1])
43
44
45 def updateARow(self,id):
46 # x.updateARow(2)
47 sql = "UPDATE food_intake SET food_had=DATE('now') WHERE id={}".format(id)
48 self.conn.execute(sql)
49 self.conn.commit()
50
51if __name__ == "__main__":
52 conn = create_database()
53 create_table(conn)
54 x = DietCounter(conn)
55
56 while True:
57 print("====================================\n")
58 userInput = int(input("1=> ADD FOOD\n2=> list Todays Food\n3=> List All FOOD EATEN:"))
59 if userInput == 1:
60 food_name = input("Use , to separate Multiple Food: ")
61 quantity = input("Use , to separate food Quantity: ")
62 x.addFood(food_name,quantity)
63 elif userInput == 2:
64 x.listAllFood('today')
65 elif userInput == 3:
66 x.listAllFood('all')
67 else:
68 break