def insert_client(column_values, row_values, id_value):
'''
Check if client has a row in Client table (aka client has filled out the
Client Profile file). If not, insert missing data into Client table.
'''
# if client did not submit client profile form
if (not (database_methods.check_id(id_value, db, "Client",
"Unique_ID_Value"))):
# extract the data and insert a row into Client table
index = [(0, 1), (2, 5), (8, 9)]
val = database_methods.fetch_values_list(row_values, index, [])
# insert into Client table
query = (
"INSERT INTO Client (Processing_Details, Unique_ID_Type, " +
"Unique_ID_Value, Date_Of_Birth, Official_Language_Preference" +
" VALUES (?, ?, ?, ?, ?, ?);")
database_methods.execute_query(query, val, 'client_data.db')
else:
# if client is in the database, update data
lst = [
"Processing_Details", "Date_Of_Birth",
"Official_Language_Preference"
]
index_list = [(0, 1), (4, 5), (8, 9)]
(col, data) = database_methods.update_lists(lst, column_values,
row_values, index_list)
query = database_methods.create_update_query(col, data, "Client",
"Unique_ID_Value",
id_value)
database_methods.update_query(query, db)
def update_agency_status(username, status):
'''
Update an agency's status.
'''
query = "Update Agency_User SET Status = ? WHERE Username = ?"
fields = (status, username)
database_methods.execute_query(query, fields, 'users.db')
def remove_temp(username):
'''
Removes the temporary password after a new one is created
'''
query = ("DELETE FROM Temp_Passcode WHERE Username = ?")
fields = (username,)
database_methods.execute_query(query, fields, 'users.db')
def insert_file_info(num, username, filename, service, month, year):
'''
Inserts a record of a file submission into the database.
'''
query = ("INSERT INTO Submission_History "
+ "values (?, ?, ?, ?, ?, ?, strftime('%s','now'))")
fields = (num, username, filename, service, month, year)
database_methods.execute_query(query, fields, 'users.db')
def update_pass(username, password):
'''
Updates the user's password with the new one that was entered
'''
hash_temp = password_hash.hash_password(password)
query = ("UPDATE User SET Password = ? WHERE Username = ?")
fields = (hash_temp, username)
database_methods.execute_query(query, fields, 'users.db')
def accepted_request(num, username):
'''
An admin accepted a request for change in data fields.
'''
status = "Accepted"
query = ("UPDATE Request SET Admin_Username = ?, Status = ?, "
"Date_Accepted = strftime('%s','now') WHERE ID = ?")
fields = (username, status, num)
database_methods.execute_query(query, fields, 'users.db')
def insert_request(num, username, desc):
'''
Insert a request from the TEQ staff into table Request.
'''
status = "Pending"
query = ("INSERT INTO Request (ID, Username, Description, Status, "
"Date_Submitted) values (?, ?, ?, ?, strftime('%s','now'))")
fields = (num, username, desc, status)
database_methods.execute_query(query, fields, 'users.db')
def insert_temp_pass(username, temp_pass):
'''
Inserts a temporary passcode into the database, or updates it if it
already exists
'''
hash_temp = password_hash.hash_password(temp_pass)
query = ("INSERT OR REPLACE INTO Temp_Passcode (Username, Temp_Pass)" +
" values (?, ?)")
fields = (username, hash_temp)
database_methods.execute_query(query, fields, 'users.db')
def delete_user(username):
'''
Deletes an agency user from the database and returns a tuple of booleans.
The first boolean determines if the account exists and the second boolean
determines if it is an agency account.
'''
query = "SELECT Username, Type from User where Username = ?"
fields = (username,)
cur = database_methods.execute_query(query, fields, 'users.db')
account = cur.fetchall()
if (account):
if (account[0][1] == "Agency"):
query = "DELETE FROM User WHERE Username = ?"
fields = (username,)
cur = database_methods.execute_query(query, fields, 'users.db')
return (True, True)
else:
return (True, False)
else:
return (False, False)
def check_email(email):
'''
Checks if an email exists in the database and returns True if it exists,
False otherwise.
'''
query = ("SELECT email from User WHERE Email = ?")
fields = (email,)
cur = database_methods.execute_query(query, fields, 'users.db')
email = cur.fetchall()
if (len(email) != 0):
return True
else:
return False
def login(username, password):
'''
Returns the name of the user and the type if login information is correct.
'''
query = ("SELECT Name, Type, Password from User where Username = ?")
fields = (username,)
cur = database_methods.execute_query(query, fields, 'users.db')
name = cur.fetchall()
if (name):
correct_password = password_hash.verify_password(password, name[0][2])
if (len(name) != 0 and correct_password):
return (name[0][0], name[0][1])
return (None, None)
def check_duplicate(username, email):
'''
Checks if there are duplicate keys which should be unique in the database.
Returns list, bool for success, username, email
'''
(conn, cur) = database_methods.connection('users.db')
query1 = "SELECT Username from User where Username = ?"
fields1 = (username,)
query2 = "SELECT Email from User where Email = ?"
fields2 = (email,)
cur1 = database_methods.execute_query(query1, fields1, 'users.db')
cur2 = database_methods.execute_query(query2, fields2, 'users.db')
usernames = cur1.fetchall()
emails = cur2.fetchall()
if (usernames and emails):
return [False, usernames[0][0], emails[0][0]]
elif (usernames):
return [False, usernames[0][0], ""]
elif (emails):
return [False, "", emails[0][0]]
else:
return [True, "", ""]
def get_username(email):
'''
(str) -> (str, str)
Getting username when given email
'''
query = ("SELECT Username, Name from User WHERE Email = ?")
fields = (email,)
cur = database_methods.execute_query(query, fields, 'users.db')
recieved = cur.fetchall()
if (len(recieved) != 0):
return (recieved[0][0], recieved[0][1])
else:
return (None, None)
def insert_user(username, email, name, password, account_type):
'''
Inserts a new user into the database (table User) and returns whether it
was successfully input or not
'''
# time is stored as UNIX timestap in INT
new_password = password_hash.hash_password(password)
query = "INSERT INTO User values (?, ?, ?, ?, ?, strftime('%s','now'));"
fields = (username, email, name, new_password, account_type)
cur = database_methods.execute_query(query, fields, 'users.db')
user = cur.fetchall()
if (user):
insert_account(username, account_type)
return True
else:
return False
def insert_row(lst, table):
'''
Inserts the values in lst into the database table.
'''
query = create_query(lst, table)
database_methods.execute_query(query, lst, 'client_data.db')
def insert_account(username, account_type):
if (account_type == "Agency"):
status = "Active"
query = "INSERT INTO Agency_User values (?, ?)"
fields = (username, status)
database_methods.execute_query(query, fields, 'users.db')
