mirror of https://github.com/stijndcl/didier
Clean up memegen code a bit
parent
8238bdf6de
commit
f2b62c3ce7
12
cogs/fun.py
12
cogs/fun.py
|
@ -94,22 +94,22 @@ class Fun(commands.Cog):
|
|||
result = memes.getMeme(name)
|
||||
|
||||
# No meme found
|
||||
if not result[0]:
|
||||
return await ctx.send(result[1])
|
||||
if result is None:
|
||||
return await ctx.send("Deze meme staat niet in de database.")
|
||||
|
||||
# Convert to list to support item assignment
|
||||
fields = list(fields)
|
||||
|
||||
# If there's only one field, the user isn't required to use quotes
|
||||
if result[1][2] == 1:
|
||||
if result[2] == 1:
|
||||
fields = [" ".join(fields)]
|
||||
|
||||
# Apply mock to mocking spongebob memes
|
||||
if result[1][1] == "mocking spongebob":
|
||||
if result[1] == "mocking spongebob":
|
||||
fields = list(map(mock.mock, fields))
|
||||
|
||||
# X, X everywhere only takes X as an argument
|
||||
if result[1][1] == "x, x everywhere":
|
||||
if result[1] == "x, x everywhere":
|
||||
fields[0] = " ".join(fields)
|
||||
fields.append(fields[0] + " everywhere")
|
||||
|
||||
|
@ -127,7 +127,7 @@ class Fun(commands.Cog):
|
|||
|
||||
if req["success"]:
|
||||
caption = {
|
||||
"template_id": result[1][0],
|
||||
"template_id": result[0],
|
||||
"username": os.getenv("IMGFLIPNAME"),
|
||||
"password": os.getenv("IMGFLIPPASSWORD"),
|
||||
"boxes[0][text]": boxes[0]["text"],
|
||||
|
|
|
@ -130,7 +130,7 @@ class ModCommands(commands.Cog):
|
|||
# Adds a meme into the database
|
||||
@add.command(name="Meme", aliases=["Mem"], usage="[Id] [Name] [Aantal Velden]")
|
||||
async def meme(self, ctx, memeid, meme, fields):
|
||||
await ctx.send(memes.insert(memeid, meme, fields)[1])
|
||||
await ctx.send(memes.insert(memeid, meme, fields))
|
||||
|
||||
# Adds a person's GitHub into the database
|
||||
@add.command(name="GitHub", aliases=["Gh", "Git"], usage="[Id] [Link]")
|
||||
|
|
|
@ -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…
Reference in New Issue