def addFriend(user_email, friend_email):
table = "Users"
with con:
cur = con.cursor()
if checkEmail(table, friend_email)[0]:
# get the users friends
user_friends = cur.execute(
"SELECT Friends FROM {} WHERE Email = '{}';".format(
table, user_email))
user_friends = user_friends.fetchone()[0]
# get new friend's ID
friendID = checkEmail(table, friend_email)[1]
# add new friend to all friends
user_friends = str(user_friends) + "_" + str(friendID)
if user_friends[0] == "_":
user_friends = user_friends[1:]
# Insert updated friends to user
cur.execute(
"UPDATE {} Set Friends = '{}' WHERE Email = '{}';".format(
table, user_friends, user_email))
return True
else:
return False
def deleteUser(email):
if checkEmail(table, email):
with con:
cur = con.cursor()
cur.execute("DELETE FROM {} WHERE Email = '{}'".format(
table, email))
return True
# should never get here
return False
def deleteFriend(user_email, friend_email):
table = "Users"
with con:
cur = con.cursor()
if checkEmail(table, friend_email)[0]:
# get the users friends
user_friends = cur.execute("SELECT Friends FROM {} WHERE Email = '{}';".format(table,user_email))
user_friends = str(user_friends.fetchone()[0])
# get new friend's ID
friendID = str(checkEmail(table,friend_email)[1])
user_friends = user_friends.split("_")
user_friends.remove(friendID)
user_friends = "_".join(user_friends)
# Insert updated friends to user
cur.execute("UPDATE {} Set Friends = '{}' WHERE Email = '{}';".format(table, user_friends, user_email))
return True
else:
return False
def addPIN(user_email, pin):
# table = "Users"
with con:
cur = con.cursor()
if checkEmail(table, user_email)[0]:
# Insert pin to
cur.execute("UPDATE {} Set PIN = {} WHERE Email = '{}';".format(
table, pin, user_email))
return True
else:
return False
def deleteAllFriends(user_email):
table = "Users"
with con:
cur = con.cursor()
if checkEmail(table, user_email)[0]:
# get the users friends
user_friends = ""
# Insert updated friends to user
cur.execute(
"UPDATE {} Set Friends = '{}' WHERE Email = '{}';".format(
table, user_friends, user_email))
return True
else:
return False
def createUser(username, password, email):
with con:
cur = con.cursor()
# find the id number
try:
num = cur.execute("SELECT MAX(Id) FROM " + str(table))
id = num.fetchone()[0] + 1
except TypeError:
id = 1
if checkEmail(table, email)[0]:
return False
insert(table, id, username, email, password, pin, phone, friends,
connectionInfo)
return True