Calculate maximum available snooker break

snooker_break.py

def max_break(reds_remaining, colour=0):
    """
    :param reds_remaining: number of reds left on table
    :param colour: (optional) point value of current colour
    :return: return maximum available snooker break
    """
    colours = sum(range(2, 8))

    if reds_remaining > 0:
        reds_and_blacks = reds_remaining * 8
        return reds_and_blacks + colours + colour

    elif reds_remaining == 0 and colour == 0:
        print "please enter current colour value"

    else:
        return sum(range(colour, 8))