bellingcat/sugartrail
1
0
Fork 0
mirror of https://github.com/bellingcat/sugartrail.git synced 2026-06-08 03:28:31 +03:00
Code Issues Packages Projects Releases Wiki Activity

Updated find_network_connections to be re-run on existing networks without re-running API calls

Browse Source
This commit is contained in:
seangreaves
2023-09-18 11:58:23 +01:00
parent 726050913b
commit 0a43de5c4b
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
sugartrail/processing.py
View File

@@ -115,15 +115,17 @@ def find_network_connections(first_network, second_network, max_depth=5, print_p
while hops < max_depth:
first_network.progress.pre_print = str(hops) + "/" + str(max_depth) + " hops completed."
second_network.progress.pre_print = str(hops) + "/" + str(max_depth) + " hops completed."
first_network.perform_hop(1, print_progress=print_progress)
second_network.perform_hop(1, print_progress=print_progress)
if first_network.n < hops:
first_network.perform_hop(1, print_progress=print_progress)
if second_network.n < hops:
second_network.perform_hop(1, print_progress=print_progress)
hops += 1
IPython.display.clear_output(wait=True)
print(str(hops) + "/" + str(max_depth) + " hops completed.")
connectors = [x for x in list(filter(first_network.graph.__contains__, second_network.graph.keys())) if x]
if connectors:
print("Found connection(s)!")
return connectors
print(str(hops) + "/" + str(max_depth) + " hops completed.")
print("No connections found.")
return