# For fileIO
import os.path

# @message - The question to display to the user.
# @default - The default if <enter> is hit.
# @accept - The list of valid responses.
# @retry - Whether to retry on a malformed input.
# Returns 'y' or 'no'
def yninput(message, default="n", accept=['yes', 'y', 'no', 'n'], retry=True):
    # TODO: Test this
    if default.lower() not in accept:
	raise TypeError('Default as %s is not in list of accepted responses.' % default)

    default_msg=" [y/N]: "
    if default.lower() == "y" or default.lower() == "yes":
	default_msg=" [Y/n]: "

    while(1):
        answer=raw_input(message + default_msg) or default.lower()
	if answer.lower() not in accept and retry:
	    print "'%s' is an invalid response.  Please try again." % answer.lower()
	    continue
        else:
            if answer.lower() == "yes" or answer.lower() == "y":
               return "y"
            return "n"

# TODO: Test with a directory returned as the answer.
def finput(message,orval):
    while(1):
        answer=raw_input(message) or orval
	if os.path.exists(answer) and os.path.isfile(answer):
	    return answer

	print "%s doesn't exist or isn't a regular file.  Try again." % answer