With first webui
This commit is contained in:
@@ -1,10 +1,40 @@
|
||||
import sqlite3
|
||||
from . import db_name
|
||||
|
||||
def get_user():
|
||||
def get_all_users():
|
||||
conn = sqlite3.connect(db_name)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('select * from T_USERS')
|
||||
rows = cursor.fetchall()
|
||||
for row in rows:
|
||||
print(row)
|
||||
print(row)
|
||||
|
||||
def get_userinfo_by_username(username):
|
||||
conn = sqlite3.connect(db_name)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('select * from T_USERS where username = ?', (username,))
|
||||
rows = cursor.fetchall()
|
||||
for row in rows:
|
||||
print(row)
|
||||
|
||||
def get_password_by_username(username):
|
||||
conn = sqlite3.connect(db_name)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('SELECT PASSWORD FROM T_USERS WHERE username = ?', (username,))
|
||||
row = cursor.fetchone()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
if row is None:
|
||||
return None
|
||||
return row[0] # password string
|
||||
|
||||
def get_password_by_email(email):
|
||||
conn = sqlite3.connect(db_name)
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('SELECT PASSWORD FROM T_USERS WHERE EMAIL = ?', (email,))
|
||||
row = cursor.fetchone()
|
||||
cursor.close()
|
||||
conn.close()
|
||||
if row is None:
|
||||
return None
|
||||
return row[0] # password string
|
||||
@@ -1,19 +1,20 @@
|
||||
from hashlib import sha512
|
||||
import sqlite3
|
||||
import useful.check as check
|
||||
import useful.hash as hash
|
||||
from . import db_name
|
||||
|
||||
def set_password_hash(password):
|
||||
return sha512(password.encode('utf-8')).hexdigest()
|
||||
|
||||
def set_login(email, password):
|
||||
|
||||
def set_login(username, email, password):
|
||||
db_con = sqlite3.connect(db_name)
|
||||
db_cur = db_con.cursor()
|
||||
try:
|
||||
if check.check_email(email):
|
||||
db_cur.execute("INSERT INTO T_USERS (USERNAME, EMAIL, PASSWORD) VALUES (?,?,?)", ('test',email, set_password_hash(password)))
|
||||
db_cur.execute("INSERT INTO T_USERS (USERNAME, EMAIL, PASSWORD) VALUES (?,?,?)", (username ,email , hash.get_password_hash(password)))
|
||||
db_con.commit()
|
||||
else:
|
||||
print("Email entered is not valid")
|
||||
except sqlite3.IntegrityError:
|
||||
print("Username or Email entered is not unique")
|
||||
print("Username or Email entered is not unique")
|
||||
db_con.close()
|
||||
Reference in New Issue
Block a user