def insert_user(values):
conn = database.connect_to_DB()
sql = '''INSERT INTO Users(Fname, Lname, Email, Psw, Salt)
VALUES(?, ?, ?, ?, ?);'''
conn.cursor().execute(sql, values)
conn.commit()
conn.close()
def tearDown(self):
conn = database.connect_to_DB()
try:
sql = "DELETE FROM Users WHERE Email = 'test@test'"
conn.cursor().execute(sql)
conn.commit()
finally:
conn.close()
def extract_user(email):
conn = database.connect_to_DB()
sql = 'SELECT * FROM Users WHERE Email = ?;'
result = conn.cursor()
result.execute(sql, (email, ))
return result.fetchone()
# that is an interesting bug
# conn.cursor().execute(sql, (email, ))
# return conn.cursor().fetchone()
conn.close()
def insert_test(self):
DB.createDB()
conn = database.connect_to_DB()
try:
sql = '''INSERT INTO Users(Fname, Lname, Email, Psw, Salt)
VALUES(?, ?, ?, ?, ?);'''
psw = pass_func.crypt_psw('test', 2)
values = ('test', 'test', 'test@test', psw, 2)
conn.cursor().execute(sql, values)
conn.commit()
finally:
conn.close()
def createDB(test_on=False):
conn = database.connect_to_DB()
# ID -> Primary key, Email, Pass, First and Last Name and Salt(needet in
# crypt algorithm
conn.cursor().execute('''CREATE TABLE IF NOT EXISTS Users
(Id INTEGER PRIMARY KEY AUTOINCREMENT,
Fname VARCHAR(128),
Lname VARCHAR(128),
Email VARCHAR(255),
Psw VARCHAR(64),
Salt TINYINT NOT NULL);''')
# ID and Key > primary
# key the key is referenced to the user which creates it
conn.cursor().execute('''CREATE TABLE IF NOT EXISTS StreamKeys
(Id INTEGER NOT NULL,
Skey VARCHAR(64),
PRIMARY KEY (Id, Skey),
FOREIGN KEY (Id) REFERENCES Users(Id)
ON DELETE CASCADE);''')
conn.close()