29 lines
925 B
Python
29 lines
925 B
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
"""MODULE: Provides functions to call various APIs to retrieve IP/prefix information."""
|
||
|
|
|
||
|
|
import ipaddress
|
||
|
|
import requests
|
||
|
|
|
||
|
|
|
||
|
|
def get_ip_information(ipv4_address: ipaddress.IPv4Address) -> dict:
|
||
|
|
"""Retrieves information about a given IPv4 address from IP-API.com."""
|
||
|
|
api_endpoint = f"http://ip-api.com/json/{ipv4_address}"
|
||
|
|
resp = requests.get(api_endpoint).json()
|
||
|
|
return resp
|
||
|
|
|
||
|
|
|
||
|
|
def get_autonomous_system_number(as_info: str) -> str:
|
||
|
|
"""Parses AS number from provided AS information."""
|
||
|
|
as_number = as_info.split(" ")[0]
|
||
|
|
return as_number
|
||
|
|
|
||
|
|
|
||
|
|
def get_prefix_information(autonomous_system: int) -> list:
|
||
|
|
"""Retrieves prefix information about a given autonomous system."""
|
||
|
|
api_endpoint = f"https://api.hackertarget.com/aslookup/?q={str(autonomous_system)}"
|
||
|
|
resp = requests.get(api_endpoint).text
|
||
|
|
prefixes = resp.split("\n")
|
||
|
|
prefixes.pop(0)
|
||
|
|
return prefixes
|