yes_no_input

yes_no_input.py

def yes_no(arg):
    """get input yes or no"""
    print '%s [y|n]:' % arg,
    if str(raw_input()).lower() == 'y':
        return True
    print 'Oops exiting!'
    return False


def get_input(msg=None, password=False):
    """get user raw_input"""
    if password:
        import getpass
        input_str = str(getpass.getpass(prompt='%s' % msg))
    else:
        if msg:
            print '%s:' % msg,
        input_str = str(raw_input())
    if input_str:
        return input_str
    else:
        raise Exception("Exiting: no input")