Skip to content

Instantly share code, notes, and snippets.

@garrettdreyfus
Last active November 18, 2024 11:01
Show Gist options
  • Save garrettdreyfus/8153571 to your computer and use it in GitHub Desktop.
Save garrettdreyfus/8153571 to your computer and use it in GitHub Desktop.
Dead simple python function for getting a yes or no answer.
def yes_or_no(question):
reply = str(raw_input(question+' (y/n): ')).lower().strip()
if reply[0] == 'y':
return True
if reply[0] == 'n':
return False
else:
return yes_or_no("Uhhhh... please enter ")
@pc-magas
Copy link

pc-magas commented Nov 18, 2024

I prefer this approach:

def yes_or_no_prompt(question:str,default_yes=True)->bool:

    inputMapping={
        'y':'y',
        "υ":'y',
        "Υ":'y',
        'n':'n',
        'ν':'n',
        'N':'n'
    }

    question = question.strip()

    if question == "":
        raise ValueError(f"Question is an empty string")

    default ='n'
    prompt_choice="(y/N)"
    if default_yes:
        default='y'
        prompt_choice="(Y/n)"

    reply = str(input(f"{question} {prompt_choice}: ")).lower().strip()
    reply = reply[0] or default
    reply = inputMapping[reply]

    if reply not in ['y','n']:
        raise ValueError(f'Invalid input for reply given {reply} instead of Y or N')

    return reply=='y'
 
 
 # Usage Example
 if __name__ == "__main__":

   while True:
     try:
        if yes_or_no_prompt("Are roses red?"):
           print("Correct")
        else:
           print("Nope")
        exit(0)
     except ValueError:
          print("I made a Boo Boo mommy")

The reason why is because I split whether I'll re-ask the question into another point whereas the value sanitization and checking is performed upon the function itself. Furthermore I set the default value as well. Furthermore I map any Greek Input given from user intop its respective English one in case keyboard layout is a non-English one.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment