Update database.py

database.py

@@ -3,6 +3,7 @@ import aiosqlite, os, bcrypt, sqlite3
 DB_PATH = os.path.join(os.getcwd(), "data", "cyber_vault.db")
 
 async def init_db():
+    if not os.path.exists(os.path.dirname(DB_PATH)): os.makedirs(os.path.dirname(DB_PATH))
     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()
@@ -10,7 +11,19 @@ async def init_db():
 def add_user(u, p):
     h = bcrypt.hashpw(p.encode(), bcrypt.gensalt()).decode()
     conn = sqlite3.connect(DB_PATH)
-    conn.execute("INSERT INTO users (u, p) VALUES (?, ?)", (u, h))
+    conn.execute("INSERT INTO users (u, p) VALUES (?, ?)", (u.upper(), h))
     conn.commit()
     conn.close()
-    
+
+def del_user(u):
+    conn = sqlite3.connect(DB_PATH)
+    conn.execute("DELETE FROM users WHERE u = ?", (u.upper(),))
+    conn.commit()
+    conn.close()
+
+def list_users():
+    conn = sqlite3.connect(DB_PATH)
+    cur = conn.execute("SELECT u FROM users")
+    users = [row[0] for row in cur.fetchall()]
+    conn.close()
+    return users