Update database.py

database.py

@@ -1,20 +1,19 @@
 import sqlite3
 from datetime import date
+import os
+
+DB_FILE = "users.db"
 
 
-# ---------------- DB CONNECTION ----------------
 def get_connection():
-    conn = sqlite3.connect("users.db", check_same_thread=False)
+    conn = sqlite3.connect(DB_FILE, check_same_thread=False)
     return conn
 
 
-# ---------------- 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,
@@ -24,7 +23,6 @@ def create_tables():
     )
     """)
 
-    # weight tracking table
     cur.execute("""
     CREATE TABLE IF NOT EXISTS weight_history(
         id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -38,9 +36,7 @@ def create_tables():
     conn.close()
 
 
-# ---------------- REGISTER USER ----------------
 def register_user(email, password):
-
     conn = get_connection()
     cur = conn.cursor()
 
@@ -51,17 +47,13 @@ def register_user(email, password):
         )
         conn.commit()
         return True
-
     except:
         return False
-
     finally:
         conn.close()
 
 
-# ---------------- LOGIN USER ----------------
 def login_user_db(email, password):
-
     conn = get_connection()
     cur = conn.cursor()
 
@@ -77,9 +69,7 @@ def login_user_db(email, password):
     return user
 
 
-# ---------------- SAVE OTP ----------------
 def save_otp(email, otp):
-
     conn = get_connection()
     cur = conn.cursor()
 
@@ -92,9 +82,7 @@ def save_otp(email, otp):
     conn.close()
 
 
-# ---------------- VERIFY OTP ----------------
 def verify_otp_db(email, otp):
-
     conn = get_connection()
     cur = conn.cursor()
 
@@ -110,9 +98,7 @@ def verify_otp_db(email, otp):
     return user
 
 
-# ---------------- SAVE WEIGHT ----------------
 def save_weight(email, weight):
-
     today = str(date.today())
 
     conn = get_connection()
@@ -127,19 +113,17 @@ def save_weight(email, weight):
     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=?",
+        "SELECT date,weight FROM weight_history WHERE email=?",
         (email,)
     )
 
-    data = cur.fetchall()
+    history = cur.fetchall()
 
     conn.close()
 
-    return data
+    return history