From 04db342391a0a429f778ccd377ddc00dbffddf8c Mon Sep 17 00:00:00 2001 From: Luke Robles Date: Sun, 10 May 2020 10:11:29 -0700 Subject: [PATCH] making the dice roll feature a little nicer by adding a color changing embed --- app/dice.py | 25 ++++++++++++++++++++----- app/dragon-bot.py | 2 +- 2 files changed, 21 insertions(+), 6 deletions(-) diff --git a/app/dice.py b/app/dice.py index fb06fdd9..0738c97a 100644 --- a/app/dice.py +++ b/app/dice.py @@ -1,13 +1,28 @@ +import discord from random import randint def roll_logic(sides): return randint(1, sides) -def roll(n, sides): - results = tuple(roll_logic(int(sides)) for _ in range(int(n))) - message = "%s\nRolling %s %s sided die\n\n%s" % (':game_die:' * int(n), n, sides, results) - message += "\n**Total:** %s " % sum(list(results)) - return message +def roll(number_of_die, sides): + results = sum(list(tuple(roll_logic(int(sides)) for _ in range(int(number_of_die))))) + + # blue + embed_color = 3447003 + if results == 1: + # Red + embed_color = 15158332 + if results >= 10: + # Green + embed_color = 3066993 + if results >= 15: + # Gold + embed_color = 15844367 + + embed = discord.Embed(description=None, color=embed_color, type="rich") + embed.set_author(name="Rolling %s %s sided die" % (number_of_die, sides)) + embed.add_field(name="**Total**", value=":game_die: %s" % results) + return embed def parse_message(message): return roll(message.split()[1], message.split()[2]) diff --git a/app/dragon-bot.py b/app/dragon-bot.py index d61179c5..331bdd78 100644 --- a/app/dragon-bot.py +++ b/app/dragon-bot.py @@ -322,7 +322,7 @@ async def on_message(message): if message.content.startswith('!roll'): await client.send_message( message.channel, - dice.parse_message(message.content) + embed=dice.parse_message(message.content) ) if message.content.startswith('!smug'):