bitbot-3.11-fork/modules/dice.py

#--depends-on commands

import random, re
from src import ModuleManager, utils

ERROR_FORMAT = "Incorrect format! Format must be [number]d[number], e.g. 1d20"
RE_DICE = re.compile("^([1-9]\d*)?d([1-9]\d*)((?:\s*[-+][1-9]\d{,2})*)\s*$",
    re.I)
RE_MODIFIERS = re.compile("([-+]\d+)")

MAX_DICE = 6
MAX_SIDES = 100

class Module(ModuleManager.BaseModule):
    @utils.hook("received.command.roll")
    @utils.hook("received.command.dice", alias_of="roll")
    @utils.kwarg("help", "Roll dice DND-style")
    @utils.spec("?<1d20>pattern(^(\d+)d(\d+)((?:\s*[-+]\d+)*))")
    def roll_dice(self, event):
        dice_count, side_count = 1, 6
        roll = "1d6"
        modifiers = []

        if event["spec"][0]:
            dice_count = int(event["spec"][0].group(1) or "1")
            side_count = int(event["spec"][0].group(2))
            roll = event["spec"][0].group(0)
            modifiers = RE_MODIFIERS.findall(event["spec"][0].group(3))

        if dice_count > 6:
            raise utils.EventError("Max number of dice is %s" % MAX_DICE)
        if side_count > MAX_SIDES:
            raise utils.EventError("Max number of sides is %s"
                % MAX_SIDES)

        results = random.choices(range(1, side_count+1), k=dice_count)

        total_n = sum(results)
        for modifier in modifiers:
            if modifier[0] == "+":
                total_n += int(modifier[1:])
            else:
                total_n -= int(modifier[1:])

        total = ""
        if len(results) > 1 or modifiers:
            total = " (total: %d)" % total_n

        results_str = ", ".join(str(r) for r in results)
        event["stdout"].write("Rolled %s and got %s%s" % (
            roll, results_str, total))