royalnet/rpgpack/commands/roll.py

import typing
import random
from royalnet.commands import *


class RollCommand(Command):
    name: str = "roll"

    description: str = "Roll a dice, from N to M (defaults to 1-100)."

    syntax = "[min] [max]"

    aliases = ["r", "random"]

    async def run(self, args: CommandArgs, data: CommandData) -> None:
        first: typing.Optional[str] = args.optional(0)
        second: typing.Optional[str] = args.optional(1)
        try:
            if second:
                minimum = int(first)
                maximum = int(second)
            elif first:
                minimum = 1
                maximum = int(first)
            else:
                minimum = 1
                maximum = 100
        except ValueError:
            if isinstance(first, str) and "d20" in first:
                raise InvalidInputError(f"Invalid value specified.\n"
                                        f"Were you looking for [c]{self.serf.prefix}dice[/c]?")
            else:
                raise InvalidInputError("Invalid value specified.")
        result = random.randrange(minimum, maximum+1)
        await data.reply(f"🎲 Dice roll [{minimum}-{maximum}]: [b]{result}[/b]")