diff --git a/cogs/remind.py b/cogs/remind.py index fe5385c..44d9bf7 100644 --- a/cogs/remind.py +++ b/cogs/remind.py @@ -20,9 +20,12 @@ class Remind(commands.Cog): Command group to remind the user of a certain thing every day. :param ctx: Discord Context """ - categories = ["Les", "Nightly"] - embed = discord.Embed(colour=discord.Colour.blue()) + rows = remind.getOrAddUser(ctx.author.id) + # TODO use a loop for this when not lazy + categories = [remind.getIcon(rows[1]) + " Nightly", remind.getIcon(rows[2]) + " Les"] + + embed = discord.Embed(colour=discord.Colour.blue()) embed.set_author(name="Remind Categorieën") embed.description = "\n".join(sorted(categories)) diff --git a/functions/database/remind.py b/functions/database/remind.py index 83dd632..da2b15c 100644 --- a/functions/database/remind.py +++ b/functions/database/remind.py @@ -41,3 +41,8 @@ def switchReminder(userid, column): return to + +# Function that returns a red or green circle depending on +# whether or not a value is True or False +def getIcon(val): + return ":green_circle:" if val else ":red_circle:"