diff --git a/Interlace/lib/core/input.py b/Interlace/lib/core/input.py index a762ee7..7d79114 100644 --- a/Interlace/lib/core/input.py +++ b/Interlace/lib/core/input.py @@ -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_'