Aqso commited on
Commit
dc862f6
·
verified ·
1 Parent(s): 97ea202

Create database.py

Browse files
Files changed (1) hide show
  1. database.py +16 -0
database.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import aiosqlite, os, bcrypt, sqlite3
2
+
3
+ DB_PATH = os.path.join(os.getcwd(), "data", "cyber_vault.db")
4
+
5
+ async def init_db():
6
+ async with aiosqlite.connect(DB_PATH) as db:
7
+ await db.execute("CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, u TEXT UNIQUE, p TEXT)")
8
+ await db.commit()
9
+
10
+ def add_user(u, p):
11
+ h = bcrypt.hashpw(p.encode(), bcrypt.gensalt()).decode()
12
+ conn = sqlite3.connect(DB_PATH)
13
+ conn.execute("INSERT INTO users (u, p) VALUES (?, ?)", (u, h))
14
+ conn.commit()
15
+ conn.close()
16
+