minor improvement

This commit is contained in:
Miroslav Stampar
2011-03-29 15:37:57 +00:00
parent d28ca5809b
commit d0861a00e2
2 changed files with 9 additions and 2 deletions

View File

@@ -1283,7 +1283,7 @@ def __useWizardInterface():
conf.data = readInput(message, default=None)
message = "[3] Injection difficulty ('--level'/'--risk') [Please choose: 1-Normal(default), 2-Medium, 3-Hard]: "
choice = readInput(message, default=1)
choice = readInput(message, default='1')
if choice == '2':
conf.risk = 2
conf.level = 3
@@ -1295,7 +1295,7 @@ def __useWizardInterface():
conf.level = 1
message = "[4] Enumeration ('--banner'/'--current-user'/...) [Please choose: 1-Basic(default), 2-Smart, 3-All]: "
choice = readInput(message, default=1)
choice = readInput(message, default='1')
if choice == '2':
conf.getBanner = True
conf.getCurrentUser = True