Simple input grabbing

input.py

#!/usr/bin/env python


def get_yn(question):
    prompt = question + " [y/n]: "
    ok = {'y', 'n'}
    answer = None
    while answer not in ok:
        answer = raw_input(prompt).strip().lower()
    return answer == 'y'


def main():
    if get_yn("My question?"):
        print "Yes"
    else:
        print "Noooooo!"

if __name__ == '__main__':
    main()