1
0
mirror of https://github.com/samba-team/samba.git synced 2025-03-20 22:50:26 +03:00

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

when asking the user to confirm an action, allow for an 'all'
choice, which will be used to allow the user to confirm all future
requests of the same type
This commit is contained in:
Andrew Tridgell 2011-07-05 12:38:31 +10:00
parent 28dbd8bbc1
commit c9497bd77f

View File

@ -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']