Skip to content

Commit

Permalink
contrib: add seeds progress indicator and remove asmap one in makesee…
Browse files Browse the repository at this point in the history
…ds script
  • Loading branch information
jonatack authored and laanwj committed Apr 26, 2022
1 parent 2b088e2 commit 7edacdc
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions contrib/seeds/makeseeds.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def parseline(line: str) -> Union[dict, None]:
sline = line.split()
if len(sline) < 11:
# line too short to be valid, skip it.
return None
return None
m = PATTERN_IPV4.match(sline[0])
sortkey = None
ip = None
Expand Down Expand Up @@ -141,10 +141,6 @@ def filterbyasn(asmap: ASMap, ips: List[Dict], max_per_asn: Dict, max_per_net: i
asn_count: Dict[int, int] = collections.defaultdict(int)

for i, ip in enumerate(ips_ipv46):
if i % 10 == 0:
# give progress update
print(f"{i:6d}/{len(ips_ipv46)} [{100*i/len(ips_ipv46):04.1f}%]\r", file=sys.stderr, end='', flush=True)

if net_count[ip['net']] == max_per_net:
# do not add this ip as we already too many
# ips from this network
Expand Down Expand Up @@ -184,8 +180,10 @@ def main():
asmap = ASMap(args.asmap)
print('Done.', file=sys.stderr)

print('Loading and parsing DNS seeds…', end='', file=sys.stderr, flush=True)
lines = sys.stdin.readlines()
ips = [parseline(line) for line in lines]
print('Done.', file=sys.stderr)

print('\x1b[7m IPv4 IPv6 Onion Pass \x1b[0m', file=sys.stderr)
print(f'{ip_stats(ips):s} Initial', file=sys.stderr)
Expand Down

0 comments on commit 7edacdc

Please sign in to comment.