blob: c808490e297f2462d924e04483dee950f79a63e4 (
about) (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
|
#--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*)((?:[-+]\d+)*)", re.I)
RE_MODIFIERS = re.compile("([-+]\d+)")
class Module(ModuleManager.BaseModule):
@utils.hook("received.command.roll", min_args=1)
def roll_dice(self, event):
"""
:help: Roll some dice, DND style
:usage: [1-5]d[1-20]
"""
match = RE_DICE.match(event["args_split"][0])
if match:
roll = match.group(0)
dice_count = int(match.group(1))
side_count = int(match.group(2))
modifiers = RE_MODIFIERS.findall(match.group(3))
if dice_count > 6:
raise utils.EventError("Max number of dice is 6")
if side_count > 20:
raise utils.EventError("Max number of sides is 20")
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:
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))
|
