diff options
Diffstat (limited to 'util/prompt.py')
| -rw-r--r-- | util/prompt.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/util/prompt.py b/util/prompt.py new file mode 100644 index 0000000..d6eb17b --- /dev/null +++ b/util/prompt.py @@ -0,0 +1,12 @@ +import sys + + +def query_yes_no(question, default): + if default == True: + prompt = " [Y/n] " + else: + prompt = " [y/N] " + + sys.stdout.write(question + prompt) + choice = input().lower() + return True if choice == "y" else (False if choice == "n" else default) |
