Update database.py

database.py

@@ -1,12 +1,15 @@
 import sqlite3
-from datetime import date
 
+
+# ---------------- DB CONNECTION ----------------
 def get_connection():
-    conn = sqlite3.connect("fitplan.db", check_same_thread=False)
+    conn = sqlite3.connect("users.db", check_same_thread=False)
     return conn
 
 
+# ---------------- CREATE TABLE ----------------
 def create_tables():
+
     conn = get_connection()
     cur = conn.cursor()
 
@@ -23,8 +26,9 @@ def create_tables():
     conn.close()
 
 
-# Register user
+# ---------------- REGISTER USER ----------------
 def register_user(email, password):
+
     conn = get_connection()
     cur = conn.cursor()
 
@@ -35,14 +39,35 @@ def register_user(email, password):
         )
         conn.commit()
         return True
+
     except:
         return False
+
     finally:
         conn.close()
 
 
-# Save OTP
+# ---------------- LOGIN USER ----------------
+def login_user_db(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 OTP ----------------
 def save_otp(email, otp):
+
     conn = get_connection()
     cur = conn.cursor()
 
@@ -55,8 +80,9 @@ def save_otp(email, otp):
     conn.close()
 
 
-# Verify OTP
-def verify_otp(email, otp):
+# ---------------- VERIFY OTP ----------------
+def verify_otp_db(email, otp):
+
     conn = get_connection()
     cur = conn.cursor()