Update database.py

database.py

@@ -1,14 +1,11 @@
 import sqlite3
 from datetime import date
 
-
-# ---------------- DATABASE CONNECTION ----------------
 def get_connection():
     conn = sqlite3.connect("fitplan.db", check_same_thread=False)
     return conn
 
 
-# ---------------- CREATE TABLES ----------------
 def create_tables():
     conn = get_connection()
     cur = conn.cursor()
@@ -22,22 +19,12 @@ def create_tables():
     )
     """)
 
-    cur.execute("""
-    CREATE TABLE IF NOT EXISTS fitness_data(
-        id INTEGER PRIMARY KEY AUTOINCREMENT,
-        user_email TEXT,
-        date TEXT,
-        weight REAL
-    )
-    """)
-
     conn.commit()
     conn.close()
 
 
-# ---------------- REGISTER USER ----------------
+# Register user
 def register_user(email, password):
-
     conn = get_connection()
     cur = conn.cursor()
 
@@ -54,54 +41,32 @@ def register_user(email, password):
         conn.close()
 
 
-# ---------------- LOGIN USER ----------------
-def login_user(email, password):
-
-    conn = get_connection()
-    cur = conn.cursor()
-
-    cur.execute(
-        "SELECT * FROM users WHERE email=? AND password=?",
-        (email, password)
-    )
-
-    user = cur.fetchone()
-
-    conn.close()
-
-    return user
-
-
-# ---------------- SAVE WEIGHT ----------------
-def save_weight(email, weight):
-
-    today = str(date.today())
-
+# Save OTP
+def save_otp(email, otp):
     conn = get_connection()
     cur = conn.cursor()
 
     cur.execute(
-        "INSERT INTO fitness_data(user_email,date,weight) VALUES (?,?,?)",
-        (email, today, weight)
+        "UPDATE users SET otp=? WHERE email=?",
+        (otp, email)
     )
 
     conn.commit()
     conn.close()
 
 
-# ---------------- GET WEIGHT HISTORY ----------------
-def get_weight_history(email):
-
+# Verify OTP
+def verify_otp(email, otp):
     conn = get_connection()
     cur = conn.cursor()
 
     cur.execute(
-        "SELECT date,weight FROM fitness_data WHERE user_email=?",
-        (email,)
+        "SELECT * FROM users WHERE email=? AND otp=?",
+        (email, otp)
     )
 
-    data = cur.fetchall()
+    user = cur.fetchone()
 
     conn.close()
 
-    return data
+    return user