Update database.py

database.py

@@ -1,40 +1,25 @@
-import aiosqlite, os, bcrypt, sqlite3
+import sqlite3, bcrypt
 
-# Path database di folder data
-DB_PATH = os.path.join(os.getcwd(), "data", "cyber_vault.db")
+DB_PATH = "cyber_vault.db"
 
 async def init_db():
-    # Buat folder data secara paksa jika belum ada
-    folder = os.path.dirname(DB_PATH)
-    if not os.path.exists(folder):
-        os.makedirs(folder, exist_ok=True)
-    
-    async with aiosqlite.connect(DB_PATH) as db:
-        await db.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, u TEXT UNIQUE, p TEXT)")
-        await db.commit()
+    with sqlite3.connect(DB_PATH) as conn:
+        conn.execute("CREATE TABLE IF NOT EXISTS users (u TEXT PRIMARY KEY, p TEXT)")
 
 def add_user(u, p):
-    h = bcrypt.hashpw(p.encode(), bcrypt.gensalt()).decode()
-    conn = sqlite3.connect(DB_PATH)
-    try:
-        conn.execute("INSERT INTO users (u, p) VALUES (?, ?)", (u.upper(), h))
-        conn.commit()
-    finally:
-        conn.close()
+    un = u.upper()
+    pw = bcrypt.hashpw(p.encode(), bcrypt.gensalt()).decode()
+    with sqlite3.connect(DB_PATH) as conn:
+        try:
+            conn.execute("INSERT INTO users (u, p) VALUES (?, ?)", (un, pw))
+            return True
+        except: return False
 
 def del_user(u):
-    conn = sqlite3.connect(DB_PATH)
-    try:
+    with sqlite3.connect(DB_PATH) as conn:
         conn.execute("DELETE FROM users WHERE u = ?", (u.upper(),))
-        conn.commit()
-    finally:
-        conn.close()
 
 def list_users():
-    conn = sqlite3.connect(DB_PATH)
-    try:
-        cur = conn.execute("SELECT u FROM users")
-        return [row[0] for row in cur.fetchall()]
-    finally:
-        conn.close()
+    with sqlite3.connect(DB_PATH) as conn:
+        return [r[0] for r in conn.execute("SELECT u FROM users").fetchall()]