diff options
author | 2020-10-05 19:22:34 +0200 | |
---|---|---|
committer | 2020-10-05 19:22:34 +0200 | |
commit | ebf41d3edd012378bb3b150c355ab726310d1681 (patch) | |
tree | 626fd0d8ef706f03602e5c631849166c2e04121f /bot/exts/evergreen/fun.py | |
parent | Rewrite roll command (diff) |
Move _get_random_die to me a separate function
- The function is only created once, instead of every time the roll command is run
Diffstat (limited to 'bot/exts/evergreen/fun.py')
-rw-r--r-- | bot/exts/evergreen/fun.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/bot/exts/evergreen/fun.py b/bot/exts/evergreen/fun.py index 09400d23..49c5d69e 100644 --- a/bot/exts/evergreen/fun.py +++ b/bot/exts/evergreen/fun.py @@ -57,17 +57,17 @@ class Fun(Cog): with Path("bot/resources/evergreen/caesar_info.json").open("r", encoding="UTF-8") as f: self._caesar_cipher_embed = json.load(f) + @staticmethod + def _get_random_die() -> str: + """Generate a random die emoji, ready to be sent on Discord.""" + die_name = f"dice_{random.randint(1, 6)}" + return getattr(Emojis, die_name) + @commands.command() async def roll(self, ctx: Context, num_rolls: int = 1) -> None: """Outputs a number of random dice emotes (up to 6).""" - def _get_random_die() -> str: - """Generate a random die emoji, ready to be sent on Discord.""" - die_name = f"dice_{random.randint(1, 6)}" - return getattr(Emojis, die_name) - - # Only support between 1 and 6 rolls if 1 <= num_rolls <= 6: - dice = " ".join(_get_random_die() for _ in range(num_rolls)) + dice = " ".join(Fun._get_random_die() for _ in range(num_rolls)) await ctx.send(dice) else: raise BadArgument("`!roll` only supports between 1 and 6 rolls.") |