Update database.py

database.py

@@ -1,4 +1,5 @@
 import sqlite3
+from datetime import date
 
 
 # ---------------- DB CONNECTION ----------------
@@ -7,12 +8,13 @@ def get_connection():
     return conn
 
 
-# ---------------- CREATE TABLE ----------------
+# ---------------- CREATE TABLES ----------------
 def create_tables():
 
     conn = get_connection()
     cur = conn.cursor()
 
+    # users table
     cur.execute("""
     CREATE TABLE IF NOT EXISTS users(
         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -22,6 +24,16 @@ def create_tables():
     )
     """)
 
+    # weight tracking table
+    cur.execute("""
+    CREATE TABLE IF NOT EXISTS weight_history(
+        id INTEGER PRIMARY KEY AUTOINCREMENT,
+        email TEXT,
+        date TEXT,
+        weight REAL
+    )
+    """)
+
     conn.commit()
     conn.close()
 
@@ -95,4 +107,39 @@ def verify_otp_db(email, otp):
 
     conn.close()
 
-    return user
+    return user
+
+
+# ---------------- SAVE WEIGHT ----------------
+def save_weight(email, weight):
+
+    today = str(date.today())
+
+    conn = get_connection()
+    cur = conn.cursor()
+
+    cur.execute(
+        "INSERT INTO weight_history(email,date,weight) VALUES (?,?,?)",
+        (email, today, weight)
+    )
+
+    conn.commit()
+    conn.close()
+
+
+# ---------------- GET WEIGHT HISTORY ----------------
+def get_weight_history(email):
+
+    conn = get_connection()
+    cur = conn.cursor()
+
+    cur.execute(
+        "SELECT date, weight FROM weight_history WHERE email=?",
+        (email,)
+    )
+
+    data = cur.fetchall()
+
+    conn.close()
+
+    return data