dd_dice

dungeon_and_dragons_dice.py

import random

# dice - simulate dungeon & dragons dice to generate random values

def dice(dicestring):
    """
    1d6 -> random integer between 1 and 6 
    2d6 -> random integer between 2 and 12
    1d20+5 -> random integer between 6 and 26
    """
    if dicestring.count("d") != 1:
        raise ValueError("dicestring does not have one 'd'")
    left, right = dicestring.split("d")
    number_of_dice = int(left)
    if ("+" in right) or ("-" in right):
        sign = "+" if "+" in right else "-"
        sides, delta = right.split(sign)
        sides = int(sides)
        delta = int(delta)
        if sign == "-":
            delta = -delta
    else:
        delta = 0
        sides = int(right)
    text = dicestring+": "
    result = 0
    for d in range(number_of_dice):
        roll = random.randint(1, sides)
        result += roll
        if d > 0:
            text+=f"+{roll}"
        else:
            text+= f"{roll}"
    result += delta
    if delta != 0:
        if delta > 0:
            text += f"+{delta}={result}"
        else:
            text += f"{delta}={result}"
    return result, text
    
if __name__ == "__main__":
    # testing
    for _ in range(10):
        print(dice("2d6+1"))