root/hodgestar/PythonCode/DmTools/windowroll.py

#!/usr/bin/env python

import random
import sys

def roll_d10():
    total, die = 0, 10
    while die == 10:
        die = random.randint(1,10)
        total += die
    return total

def roll_thewindow(die,add=0):
    assert(die >= 2)
    dice = []

    result = random.randint(1,die)
    dice.append(result)
        
    if result == 1:
        crit_die = random.randint(1,30)
        dice.append("d30: %d" % crit_die)
        if crit_die <= die:
            dice.append("crit")
    elif result == die:
        botch_die = random.randint(1,30)
        dice.append("d30: %d" % botch_die)
        if botch_die <= die:
            dice.append("botch")

    return result + add, dice

def to_int(s,errmsg):
    try:
        return int(s)
    except Exception, e:
        raise ValueError(errmsg)

def parse_roll(s):
    a = s.split(':')
    if len(a) == 1:
        rolls = 1
        s = a[0]
    elif len(a) == 2:
        rolls = to_int(a[0],"Number of rolls to make should be an integer.")
        s = a[1]
    else:
        raise ValueError("Unrecognised roll format.")

    a = s.split('+')
    sign = 1
    if len(a) == 1:
        a = s.split('-')
        sign = -1
    
    if len(a) == 1:
        add = 0
    elif len(a) == 2:
        add = sign * to_int(a[1],"Amount to add to or subtract from roll should be an integer.")
    else:
        raise ValueError("Unrecognised roll format.")

    die = a[0]
    if not die.startswith('d'):
        raise ValueError("Unrecognised roll format.")
    die = to_int(die[1:],"Die should be an integer.")

    return die, add, rolls

def ask_and_roll(fLog):
    try:
        sIn = raw_input(">").lower()
    except BaseException, e:
        print
        return False
    if sIn == 'q':
        return False
    else:
        try:
            if fLog:
                fLog.write(">%s\n" % sIn)

            die, add, rolls = parse_roll(sIn)

            for i in range(rolls):
                total, dice = roll_thewindow(die,add)
                print "%i: %s" % (total, ", ".join([str(x) for x in dice]))

            if fLog:
                fLog.write("%i: %s\n" % (total, ", ".join([str(x) for x in dice])))
        except Exception, e:
            print e
            print "Enter rolls as W:dX+Y."
            if fLog:
                fLog.write("%s\n" % e)
        return True

if __name__ == "__main__":
    if len(sys.argv) == 2:
        fLog = file(sys.argv[1],"a")
    else:
        fLog = None

    print "Enter rolls as W:dX+Y (W rolls of dX+Y; W: and +Y optional; can also use -Y)"
    print "Q or q to quit."
    while ask_and_roll(fLog):
        pass

    if fLog:
        fLog.close()