1
0
mirror of https://github.com/samba-team/samba.git synced 2025-07-19 04:59:10 +03:00

s4-sambatool: extract the confirm function in a separte module for reuse

This commit is contained in:
Matthieu Patou
2011-06-19 23:09:59 +04:00
committed by Matthieu Patou
parent bc549575dd
commit 9c94943d26
2 changed files with 36 additions and 10 deletions

View File

@ -20,6 +20,7 @@
import ldb, sys
import samba.getopt as options
from samba.common import confirm
from samba.auth import system_session
from samba.samdb import SamDB
from samba.netcmd import (
@ -28,14 +29,6 @@ from samba.netcmd import (
Option
)
def confirm(self, msg):
'''confirm an action with the user'''
if self.yes:
print("%s [YES]" % msg)
return True
v = raw_input(msg + ' [y/N] ')
return v.upper() in ['Y', 'YES']
class cmd_dbcheck(Command):
"""check local AD database for errors"""
synopsis = "dbcheck <DN> [options]"
@ -98,7 +91,7 @@ class cmd_dbcheck(Command):
print("ERROR: Empty attribute %s in %s" % (attrname, dn))
if not self.fix:
return
if not confirm(self, 'Remove empty attribute %s from %s?' % (attrname, dn)):
if not confirm('Remove empty attribute %s from %s?' % (attrname, dn), self.yes):
print("Not fixing empty attribute %s" % attrname)
return
@ -156,7 +149,7 @@ class cmd_dbcheck(Command):
mod_list.append((val, normalised[0]))
if not self.fix:
return
if not self.confirm(self, 'Fix normalisation for %s from %s?' % (attrname, dn)):
if not confirm('Fix normalisation for %s from %s?' % (attrname, dn), self.yes):
print("Not fixing attribute %s" % attrname)
return