mirror of
https://github.com/stijndcl/didier.git
synced 2026-04-07 15:48:29 +02:00
Clean up memegen code a bit
This commit is contained in:
parent
8238bdf6de
commit
f2b62c3ce7
4 changed files with 14 additions and 12 deletions
|
|
@ -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():
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue