Files
AS_Network_List/get_info_from_ripe.py
2024-03-03 12:45:49 +01:00

52 lines
1.7 KiB
Python
Executable File

#!/usr/bin/env python3
import argparse
import requests
from pylib.whois import whois_query
url = "https://stat.ripe.net/data/country-resource-list/data.json?resource=RU&v4_format=prefix"
def get_data(json, file, attr, field, prefix=""):
limit = args.limit
count = 0
with open(file, 'w') as f:
for x in json['data']['resources'][attr]:
count += 1
x = prefix+x.strip()
if count > limit:
response = None
else:
response = whois_query(x, field)
if response is None:
name = "-no-description-"
else:
name = response.split(':')[1].strip()
print(f"{x} {name}")
f.write(str(x+" "+name) + '\n')
parser = argparse.ArgumentParser()
parser.add_argument('--asn', action='store_true', help='Run the ASN query')
parser.add_argument('--ipv4', action='store_true', help='Run the IPv4 query')
parser.add_argument('--ipv6', action='store_true', help='Run the IPv6 query')
parser.add_argument('--all', action='store_true', help='Run all queries')
parser.add_argument('--limit', type=int, help='Limit the number of whois queries to prevent blacklisting from whois servers', default=2500)
args = parser.parse_args()
if not (args.asn or args.ipv4 or args.ipv6 or args.all):
parser.print_help()
exit()
response = requests.get(url)
response.raise_for_status()
data = response.json()
if args.asn or args.all:
get_data(data, 'auto/all-ru-asn.txt', 'asn', 'as-name', "AS")
if args.ipv4 or args.all:
get_data(data, 'auto/all-ru-ipv4.txt', 'ipv4', 'netname')
if args.ipv6 or args.all:
get_data(data, 'auto/all-ru-ipv6.txt', 'ipv6', 'netname')