mirror of https://github.com/stijndcl/didier
49 lines
1.1 KiB
Python
49 lines
1.1 KiB
Python
from functions.database import utils
|
|
|
|
|
|
def getAllRows():
|
|
connection = utils.connect()
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("SELECT * FROM remind")
|
|
return cursor.fetchall()
|
|
|
|
|
|
def getOrAddUser(userid):
|
|
connection = utils.connect()
|
|
cursor = connection.cursor()
|
|
|
|
cursor.execute("SELECT * FROM remind WHERE userid = %s", (int(userid),))
|
|
res = cursor.fetchall()
|
|
|
|
if not res:
|
|
cursor.execute("INSERT INTO remind(userid) VALUES(%s)", (int(userid),))
|
|
connection.commit()
|
|
|
|
return getOrAddUser(userid)
|
|
|
|
return res[0]
|
|
|
|
|
|
def switchReminder(userid, column):
|
|
connection = utils.connect()
|
|
cursor = connection.cursor()
|
|
|
|
columns = ["id", "nightly", "les"]
|
|
|
|
res = getOrAddUser(userid)
|
|
|
|
# Switch the column value
|
|
to = not (res[columns.index(column)])
|
|
|
|
cursor.execute("UPDATE remind SET {} = %s WHERE userid = %s".format(column), (to, int(userid),))
|
|
connection.commit()
|
|
|
|
return to
|
|
|
|
|
|
# Function that returns a red or green circle depending on
|
|
# whether or not a value is True or False
|
|
def getIcon(val):
|
|
return ":green_circle:" if val else ":red_circle:"
|