[op-mode] T1607 rewrite 'reset conntrack', 'reset & show ip[v6]' to python/xml syntax

1 files changed, 15 insertions, 0 deletions

diff --git a/python/vyos/util.py b/python/vyos/util.py
index 6ab606983..88bb0c8f4 100644
--- a/python/vyos/util.py
+++ b/python/vyos/util.py
@@ -18,6 +18,7 @@ import re
 import grp
 import time
 import subprocess
+import sys
 
 import psutil
 
@@ -176,3 +177,17 @@ def wait_for_commit_lock():
     while commit_in_progress():
         time.sleep(1)
 
+def ask_yes_no(question, default=False) -> bool:
+    """Ask a yes/no question via input() and return their answer."""
+    default_msg = "[Y/n]" if default else "[y/N]"
+    while True:
+        sys.stdout.write("%s %s " % (question, default_msg))
+        c = input().lower()
+        if c == '':
+            return default
+        elif c in ("y", "ye", "yes"):
+            return True
+        elif c in ("n", "no"):
+            return False
+        else:
+            sys.stdout.write("Please respond with yes/y or no/n\n")