Commit 00ba44cd authored by Tharushika P.R's avatar Tharushika P.R

Added DB file

parent f2528afc
import sqlite3
import model
class DB:
def __init__(self):
self._conn = sqlite3.connect('database.db', check_same_thread=False)
self._c = self._conn.cursor()
# run this method only once
def create_table_user(self):
self._c.execute("""CREATE TABLE users(
id integer PRIMARY KEY,
name text,
email text,
password text,
user_type text
)""")
self._conn.commit()
def login(self, email, password):
statement = f"SELECT * FROM users WHERE email='{email}' AND password = '{password}';"
self._c.execute(statement)
user = self._c.fetchone()
if not user: # An empty result evaluates to False.
print("Login failed")
return False, 'user'
else:
print("Welcome")
login_user = model.User(user[0], user[1], user[2], user[3], user[4])
return True, login_user
def get_user_by_email(self, email):
statement = f"SELECT * FROM users WHERE email='{email}'"
self._c.execute(statement)
users = self._c.fetchall()
returnUserList = []
for i in range(0, len(users)):
returnUserList.append(model.User(users[i][0], users[i][1], users[i][2], users[i][3], users[i][4]))
return returnUserList
def select_user(self):
self._c.execute("SELECT * FROM users")
users = self._c.fetchall()
returnUserList = []
for i in range(0, len(users)):
returnUserList.append(model.User(users[i][0], users[i][1], users[i][2], users[i][3], users[i][4]))
return returnUserList
def save_user(self, name, email, password, user_type):
sql = "INSERT INTO users (name, email, password, user_type) " \
"VALUES (?, ?, ?, ?)"
val = (name, email, password, user_type)
self._c.execute(sql, val)
self._conn.commit()
def update_user(self, id, name, email, password, user_type):
sql = "UPDATE users SET name = ?, email = ?, password = ?, user_type = ? WHERE id = ?"
val = (name, email, password, user_type, id)
self._c.execute(sql, val)
self._conn.commit()
def delete_user(self, id):
sql = 'DELETE FROM users WHERE id = ?'
val = (id)
self._c.execute(sql, val)
self._conn.commit()
# db = DB()
# print(db.get_user_by_email('sankamadushanka78@gmail.com')[0].get_name())
# db.create_table_user()
# db.save_user('Admin', 'admin@admin.lk', 'admin', 'admin')
# print(db.select_user()[0].get_email())
# print(db.select_user()[0].get_password())
# print(db.login('rs', 'll'))
# x = db.select_mark()
#
# for i in x:
# print(i.get_id())
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment