Skip to content

Commit

Permalink
Correctly parameterize variables implemented in 466e00b
Browse files Browse the repository at this point in the history
  • Loading branch information
ignis-sec committed May 19, 2021
1 parent 0d35a04 commit 6afd00a
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion Markopy/src/CLI/markopy_cli.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,8 @@
parser.add_argument("-d", "--dataset", help="Dataset file to read input from for training. Will be ignored for generation mode.")
parser.add_argument("-s", "--seperator",help="Seperator character to use with training data.(character between occurrence and value)")
parser.add_argument("-w", "--wordlist", help="Wordlist file path to export generation results to. Will be ignored for training mode")
parser.add_argument("--min", default=6, help="Minimum length that is allowed during generation")
parser.add_argument("--max", default=12,help="Maximum length that is allowed during generation")
parser.add_argument("-n", "--count", help="Number of lines to generate. Ignored in training mode.")
parser.add_argument("-v", "--verbosity",action="count", help="Output verbosity.")
args = parser.parse_args()
Expand Down Expand Up @@ -89,7 +91,7 @@ def cli_generate(model):
if(os.path.isfile(args.wordlist)):
logging.pprint(f"{args.wordlist} exists and will be overwritten.", 1)

model.Generate(int(args.count), args.wordlist,6,12)
model.Generate(int(args.count), args.wordlist, args.min, args.max)



Expand Down

0 comments on commit 6afd00a

Please sign in to comment.