fix length check

This commit is contained in:
Z. Cliffe Schreuders
2021-11-18 22:31:29 +00:00
parent e3d2e01a11
commit cef1428c64

View File

@@ -10,8 +10,8 @@ class SanitisedWordGenerator < StringGenerator
def initialize
super
self.wordlist = []
self.min_length = ''
self.max_length = ''
self.min_length = 0
self.max_length = 999
self.module_name = 'Random Word Generator'
end
@@ -27,15 +27,11 @@ class SanitisedWordGenerator < StringGenerator
when '--wordlist'
self.wordlist << arg;
when '--min_length'
if arg == ''
self.min_length = 0
else
if arg != ''
self.min_length = arg.to_i
end
when '--max_length'
if arg == ''
self.max_length = 999
else
if arg != ''
self.max_length = arg.to_i
end
end
@@ -64,4 +60,4 @@ class SanitisedWordGenerator < StringGenerator
end
end
SanitisedWordGenerator.new.run
SanitisedWordGenerator.new.run