diff options
| -rw-r--r-- | source4/scripting/python/samba/common.py | 12 |
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'] |