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 | |
| 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')
| -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.")  |