mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-27 13:34:28 +00:00
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
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.interface.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]")
|