aboutsummaryrefslogtreecommitdiffstats
path: root/bot/exts/evergreen/fun.py
diff options
context:
space:
mode:
authorGravatar Gustav Odinger <[email protected]>2020-10-05 19:22:34 +0200
committerGravatar Gustav Odinger <[email protected]>2020-10-05 19:22:34 +0200
commitebf41d3edd012378bb3b150c355ab726310d1681 (patch)
tree626fd0d8ef706f03602e5c631849166c2e04121f /bot/exts/evergreen/fun.py
parentRewrite 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.py14
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.")