Module realm_bot.roll.parse

Parse segments from roll command arguments

Global variables

var addl_roll_regex

Regex pattern for additional modifier rolls/constants

var roll_regex

Regex pattern for initial roll

Functions

def parse_segments(roll: str) ‑> list[list[RollSegment]]

Expand source code

def parse_segments(roll: str) -> list[list[RollSegment]]:
    """Parse a roll formula string into a list of RollSegment objects."""

    segments = []

    # strip all whitespace
    roll = roll.replace(" ", "")

    # parse first roll
    first = roll_regex.match(roll)
    assert first
    first = first.groupdict()

    # parse variable number of mods
    mods = [r.groupdict() for r in addl_roll_regex.finditer(roll)]

    batch = int(first["batch"] or 1)

    for m in mods:
        if m["batch"]:
            batch = int(m["batch"])

    batches = BatchRoll()

    for _ in range(batch):
        segments: list[RollSegment] = [
            DiceRoll(
                raw=first["all"],
                dice=int(first["num"] or "1"),
                faces=int(first["die"]),
                extra=first["fun"],
            )
        ]

        for mod in mods:
            if mod["die"]:
                segments.append(
                    DiceRoll(
                        raw=mod["all"],
                        negative=mod["op"] == "-",
                        dice=int(mod["num"] or "1"),
                        faces=int(mod["die"]),
                        extra=mod["fun"],
                    )
                )
            else:
                segments.append(
                    ConstantModifier(
                        raw=mod["all"],
                        negative=mod["op"] == "-",
                        number=int(mod["num"]),
                    )
                )

        batches.append(segments)

    return batches

Parse a roll formula string into a list of RollSegment objects.