Added in proto variable

This commit is contained in:
ProDigySML
2019-01-31 23:41:34 -08:00
parent 7b9ea75629
commit 22ecf4d9a1

View File

@@ -190,6 +190,13 @@ class InputHelper(object):
if arguments.output:
final_commands = InputHelper._replace_variable_for_commands(final_commands, "_output_", [arguments.output])
if arguments.proto:
if "," in arguments.proto:
protocols = arguments.proto.split(",")
else:
protocols = arguments.proto
final_commands = InputHelper._replace_variable_for_commands(final_commands, "_proto_", protocols)
return final_commands
@@ -277,6 +284,11 @@ class InputParser(object):
help='Specify a port variable that can be used in commands as _port_'
)
parser.add_argument(
'--proto', dest='proto',
help='Specify protocols that can be used in commands as _proto_'
)
parser.add_argument(
'-rp', dest='realport',
help='Specify a real port variable that can be used in commands as _realport_'