michael/Spoofy
1
0
Fork 0
mirror of https://github.com/MattKeeley/Spoofy.git synced 2026-02-03 13:33:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: threading bug

Browse Source
This commit is contained in:
Matt Keeley
2023-04-05 15:35:53 -07:00
parent 968e5318a1
commit 5731e1f834
1 changed files with 2 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
spoofy.py
View File

@@ -45,17 +45,10 @@ def process_domains(domains, output):
"""
This function is for multithreading woot woot!
"""
num_threads = os.cpu_count()
threads = []
num_domains = len(domains)
num_threads = min(num_threads, num_domains)
for i in range(num_threads):
start = i * num_domains // num_threads
end = (i + 1) * num_domains // num_threads
thread = threading.Thread(
target=process_domains_worker, args=(domains[start:end], output))
for domain in domains:
thread = threading.Thread(target=process_domain, args=(domain, output))
thread.start()
threads.append(thread)
@@ -63,11 +56,6 @@ def process_domains(domains, output):
thread.join()
def process_domains_worker(domains, output):
for domain in domains:
process_domain(domain, output)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group()