s4-pycommon: allow an optional 'all' choice for confirm dialogs

[kai/samba.git] / source4 / scripting / python / samba / common.py

diff --git a/source4/scripting/python/samba/common.py b/source4/scripting/python/samba/common.py
index a2a49627972d35488e7a79b8e76404f0ced5272e..ebb3f88733079eb31fc9d2dff392cc96062e3af3 100644 (file)
--- a/source4/scripting/python/samba/common.py
+++ b/source4/scripting/python/samba/common.py
@@ -18,7 +18,7 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #
 
-def confirm(msg, forced = False):
+def confirm(msg, forced = False, allow_all=False):
     """confirm an action with the user
         :param msg: A string to print to the user
         :param forced: Are the answer forced
@@ -27,7 +27,13 @@ def confirm(msg, forced = False):
         print("%s [YES]" % msg)
         return True
 
-    v = raw_input(msg + ' [y/N] ')
-    return v.upper() in ['Y', 'YES']
+    if allow_all:
+        v = raw_input(msg + ' [y/N/all] ')
+        if v.upper() == 'ALL':
+            return "ALL"
+        return v.upper() in ['Y', 'YES']
+    else:
+        v = raw_input(msg + ' [y/N] ')
+        return v.upper() in ['Y', 'YES']