blob: 8135b0b2b3308a41aae52b4dae521f3ff63fe92d (
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
#!/usr/bin/env python3
import random
import sys
from pathlib import Path
import re
def roll_die(num_dice: int, dice_type: int, modifier: int) -> [int]:
rolls = []
for i in range(num_dice):
rolls.append(random.randint(1, dice_type) + modifier)
return rolls
def parse_dice_notation(dice_notation) -> (int, int, int):
# Define the regular expression pattern
pattern = re.compile(r'(\d+)d(\d+)([+-]\d+)?')
# Search the pattern in the dice notation string
match = pattern.match(dice_notation)
if not match:
raise ValueError(f"Invalid dice notation: {dice_notation}")
# Extract the groups from the match
groups = match.groups()
# Convert the groups to integers and handle the optional modifier
num_dice = int(groups[0])
dice_type = int(groups[1])
modifier = int(groups[2]) if groups[2] else 0
return num_dice, dice_type, modifier
def summary(rolls: [int]) -> None:
print(rolls)
print(f'Min: {min(rolls)}')
print(f'Max: {max(rolls)}')
if len(rolls) > 1:
print(f'Total: {sum(rolls)}')
print(f'Avg: {sum(rolls)/len(rolls):.2f}')
def main():
if len(sys.argv) != 2:
path = Path(sys.argv[0]).parts[-1]
print(f"Usage: {path} xdy[(+|y)z]")
print("Rolls x number of y sided die, optionally using a modifier z")
print()
print("Examples:")
print("\t2d4+2 Rolls two four sided die adding 2 to each roll")
print("\t1d20-1 Rolls a single twenty and subtracts 1 from the roll")
sys.exit(1)
dice_notation = sys.argv[1]
try:
num_die, die_type, modifier = parse_dice_notation(dice_notation)
except ValueError as e:
print(e)
rolls = roll_die(num_die, die_type, modifier)
summary(rolls)
if __name__ == "__main__":
main()
|
