mirror of https://github.com/stijndcl/didier
28 lines
995 B
Python
28 lines
995 B
Python
|
from functions.database import utils
|
||
|
|
|
||
|
|
|
||
|
|
def get_user(userid):
|
||
|
|
connection = utils.connect()
|
||
|
|
cursor = connection.cursor()
|
||
|
|
cursor.execute("SELECT day, month, year FROM birthdays WHERE userid = %s", (int(userid),))
|
||
|
|
return cursor.fetchall()
|
||
|
|
|
||
|
|
|
||
|
|
def get_users_on_date(day, month):
|
||
|
|
connection = utils.connect()
|
||
|
|
cursor = connection.cursor()
|
||
|
|
cursor.execute("SELECT userid FROM birthdays WHERE day = %s AND month = %s", (int(day), int(month),))
|
||
|
|
return cursor.fetchall()
|
||
|
|
|
||
|
|
|
||
|
|
def add_user(userid, day, month, year):
|
||
|
|
connection = utils.connect()
|
||
|
|
cursor = connection.cursor()
|
||
|
|
if get_user(userid):
|
||
|
|
cursor.execute("UPDATE birthdays SET day = %s, month = %s, year = %s WHERE userid = %s",
|
||
|
|
(int(day), int(month), int(year), int(userid),))
|
||
|
|
else:
|
||
|
|
cursor.execute("INSERT INTO birthdays(userid, day, month, year) VALUES (%s, %s, %s, %s)",
|
||
|
|
(int(userid), int(day), int(month), int(year),))
|
||
|
|
connection.commit()
|