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

Serious Speed Improvements for Large Lists

Browse Source
This commit is contained in:
ProDigySML
2019-03-24 21:50:20 -07:00
parent ec93f84d5d
commit b94f62ac73
1 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -5,6 +5,7 @@ import os.path
from os import access, W_OK from os import access, W_OK
import sys import sys
from re import compile from re import compile
from random import sample
class InputHelper(object): class InputHelper(object):
@@ -63,10 +64,16 @@ class InputHelper(object):
def _replace_variable_for_commands(commands, variable, replacements): def _replace_variable_for_commands(commands, variable, replacements):
tmp_commands = set() tmp_commands = set()
test = list()
if not variable in sample(commands, 1)[0]:
return commands
for replacement in replacements: for replacement in replacements:
for command in commands: for command in commands:
tmp_commands.add(str(command).replace(variable, str(replacement))) test.append(str(command).replace(variable, str(replacement)))
tmp_commands.update(test)
return tmp_commands return tmp_commands
@@ -111,6 +118,7 @@ class InputHelper(object):
# process targets first # process targets first
if arguments.target: if arguments.target:
ranges.add(arguments.target) ranges.add(arguments.target)
print("List made")
else: else:
targetFile = arguments.target_list targetFile = arguments.target_list
if not sys.stdin.isatty(): if not sys.stdin.isatty():
@@ -130,8 +138,10 @@ class InputHelper(object):
# 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:
target = target.replace(" ", "") target = target.replace(" ", "")
print("Fixing spaces")
for ips in target.split(","): for ips in target.split(","):
# check if it is a domain name # check if it is a domain name
if ips.split(".")[-1][0].isalpha(): if ips.split(".")[-1][0].isalpha():
targets.add(ips) targets.add(ips)
@@ -147,6 +157,7 @@ class InputHelper(object):
targets.update(InputHelper._get_ips_from_glob(ips)) targets.update(InputHelper._get_ips_from_glob(ips))
else: else:
targets.add(ips) targets.add(ips)
print("added")
# 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 exclusion in exclusions_ranges: for exclusion in exclusions_ranges:
@@ -169,6 +180,7 @@ class InputHelper(object):
else: else:
exclusions.add(ips) exclusions.add(ips)
print("removed")
# difference operation # difference operation
targets -= exclusions targets -= exclusions