aboutsummaryrefslogtreecommitdiffstats
path: root/util
diff options
context:
space:
mode:
Diffstat (limited to 'util')
-rw-r--r--util/prompt.py12
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)