Das deutsche Python-Forum

Code: Alles auswählen

import random

def d(n, aces=False):
    r = random.randrange(n) + 1
    return r + d(n, aces) if aces and r == n else r

Code: Alles auswählen

import re
_RE_DICE = re.compile(r"(\d+)?d(\d+)?|(\d+)|([-+])")

def roll(dice, aces=False):
    result = 0; sign = 1
    for m in _RE_DICE.finditer(dice):
        if m.lastindex == 2:
            count = int(m.group(1) or 1)
            sides = int(m.group(2) or 6)
            for i in range(count):
                result += d(sides, aces) * sign
        elif m.lastindex == 3:
            result += int(m.group(3)) * sign
        else:
            sign = +1 if m.group(4) == '+' else -1
    return result

Code: Alles auswählen

import re
import random
import operator


class Bunch(object):
    def __init__(self, **kwargs):
        self.__dict__.update(kwargs)


match = re.compile(
    flags = re.VERBOSE,
    pattern = r"""
        ( (?P<times> \d* )
          (?P<item> d | D )
          (?P<sides> \d* )
          ( (?P<op> \+ | \- )(?P<value> \d+ ) | $ )
          |
          (?P<error> .* )
        )
    """
).match


ops = {'+':operator.add, '-':operator.sub, None:lambda x,_:x}


def roll(dice):
    parsed = Bunch(**match(dice).groupdict())
    if parsed.error:
        raise ValueError('Invalid argument "%s"' % parsed.error)
    times = int(parsed.times or 1)
    sides = int(parsed.sides or 6)
    value = int(parsed.value or 0)
    op = ops[parsed.op]
    return max(
        0, op(sum(random.randint(1, sides) for each in xrange(times)), value)
    )

print roll('12d8-3')
print roll('12d8')
print roll('12d-3')
print roll('12d')
print roll('d19+100')
print roll('d19')
print roll('d')