aljaz/Interlace
1
1
Fork 0
mirror of https://github.com/codingo/Interlace.git synced 2025-12-17 23:04:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fixed the errors for ports and exclusions

Browse Source
This commit is contained in:
ProDigySML
2019-01-10 03:02:19 -08:00
parent 6a0b1d800b
commit 30e6b71c36
1 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
Interlace/lib/core/input.py
View File

@@ -66,13 +66,14 @@ class InputHelper(object):
final_commands = set() final_commands = set()
output = OutputHelper(arguments) output = OutputHelper(arguments)
if "," in arguments.port: if arguments.port:
ports = arguments.port.split(",") if "," in arguments.port:
elif "-" in arguments.port: ports = arguments.port.split(",")
tmp_ports = arguments.port.split("-") elif "-" in arguments.port:
ports = list(range(int(tmp_ports[0]), int(tmp_ports[1]) + 1)) tmp_ports = arguments.port.split("-")
else: ports = list(range(int(tmp_ports[0]), int(tmp_ports[1]) + 1))
ports = [arguments.port] else:
ports = [arguments.port]
# process targets first # process targets first
@@ -86,8 +87,9 @@ class InputHelper(object):
if arguments.exclusions: if arguments.exclusions:
exclusions_ranges.add(arguments.exclusions) exclusions_ranges.add(arguments.exclusions)
else: else:
for exclusion in arguments.exclusions_list: if arguments.exclusions_list:
exclusions_ranges.add(target.strip()) for exclusion in arguments.exclusions_list:
exclusions_ranges.add(target.strip())
# removing elements that may have spaces (helpful for easily processing comma notation) # removing elements that may have spaces (helpful for easily processing comma notation)
for target in ranges: for target in ranges: