Clean up memegen code a bit

This commit is contained in:
Stijn De Clercq 2021-01-25 23:21:40 +01:00
parent 8238bdf6de
commit f2b62c3ce7
4 changed files with 14 additions and 12 deletions

View file

@ -2,13 +2,15 @@ from functions.database import utils
def insert(id, name, fields):
if getMeme(name)[0]:
return [False, "Deze meme staat al in de database."]
if getMeme(name) is not None:
return "Deze meme staat al in de database."
connection = utils.connect()
cursor = connection.cursor()
cursor.execute("INSERT INTO memes(id, name, fields) VALUES (%s, %s, %s)", [int(id), name.lower(), int(fields)])
connection.commit()
return [True, "{} is toegevoegd aan de database.".format(name[0].upper() + name[1:].lower())]
return "{} is toegevoegd aan de database.".format(name[0].upper() + name[1:].lower())
def getMeme(name):
@ -17,8 +19,8 @@ def getMeme(name):
cursor.execute("SELECT * FROM memes WHERE name like %s", ["%" + name.lower() + "%"])
result = cursor.fetchall()
if len(result) == 0:
return [False, "Deze meme staat niet in de database."]
return [True, result[0]]
return None
return result[0]
def getAllMemes():

0
functions/memes.py Normal file
View file