Create reminder task & command, fix Les aliases

This commit is contained in:
Stijn De Clercq 2020-10-23 21:34:10 +02:00
parent caf595010b
commit 250c2d40c7
6 changed files with 133 additions and 4 deletions

View file

@ -0,0 +1,43 @@
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 = %s WHERE userid = %s", (column, to, int(userid),))
connection.commit()
return to