root/hodgestar/PythonCode/DmTools/l5rroll.py

#!/usr/bin/env python

import random
import sys

try:
    # use readline if we can for nice line navigation
    # with raw_input
    import readline
except ImportError:
    pass


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

def roll_l5r(roll,keep,add=0):
    dice = [roll_d10() for i in range(roll)]
    dice.sort()
    dice.reverse()
    kept = dice[:keep]
    return sum(kept) + 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('+')
    if len(a) == 1:
        add = 0
    elif len(a) == 2:
        add = to_int(a[1],"Amount to add to roll should be an integer.")
    else:
        raise ValueError("Unrecognised roll format.")
    s = a[0]

    a = s.split('k')
    if len(a) != 2:
        raise ValueError("Unrecognised roll format.")

    roll = to_int(a[0],"Number of dice to roll should be an integer.")
    keep = to_int(a[1],"Number of dice to keep should be an integer.")

    return roll, keep, 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)

            roll, keep, add, rolls = parse_roll(sIn)

            for i in range(rolls):
                total, dice = roll_l5r(roll,keep,add)
                print "%i: %s" % (total, dice)

            if fLog:
                fLog.write("%i: %s\n" % (total, dice))
        except Exception, e:
            print e
            print "Enter rolls as W:XkY+Z."

            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:XkY+Z (W rolls of XkY+Z; W: and +Z optional)"
    print "Q or q to quit."
    while ask_and_roll(fLog):
        pass

    if fLog:
        fLog.close()