diff --git a/scripts/roa.py b/scripts/roa.py index db92a5b..67634a1 100755 --- a/scripts/roa.py +++ b/scripts/roa.py @@ -193,12 +193,10 @@ if __name__ == "__main__": as_route4 = list() as_route6 = list() vkeys = [k for k in VALID_KEYS if k != 'asn'] - for r in roa4: - if r['asn'] == asn: - as_route4.append({k:v for k, v in r.items() if k in vkeys}) - for r in roa6: - if r['asn'] == asn: - as_route6.append({k:v for k, v in r.items() if k in vkeys}) + for roa, as_route in ((roa4, as_route4), (roa6, as_route6)): + for r in roa: + if r['asn'] == asn: + as_route.append({k:v for k, v in r.items() if k in vkeys}) owner = asi['owner'] peopledict = d_output['people'].setdefault(owner, {"info": PEOPLE[owner], "asns": list()}) peopledict['asns'].append({"asn": asn, "routes": {'ipv4': as_route4, 'ipv6': as_route6}})