summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--source4/scripting/python/samba/common.py12
1 files changed, 9 insertions, 3 deletions
diff --git a/source4/scripting/python/samba/common.py b/source4/scripting/python/samba/common.py
index a2a4962797..ebb3f88733 100644
--- 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']