Luke Tainton
af8fcad53a
* Added checks to ensure API responses are as expected * Add tests for broken IP/prefix API responses * Add tests for bad IP/prefix status codes
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
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}"
|
|
try:
|
|
resp = requests.get(api_endpoint)
|
|
resp.raise_for_status()
|
|
ret = resp.json() if resp.json().get("status") == "success" else None
|
|
except (requests.exceptions.JSONDecodeError, requests.exceptions.HTTPError):
|
|
ret = None
|
|
return ret
|
|
|
|
|
|
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)}"
|
|
try:
|
|
resp = requests.get(api_endpoint)
|
|
resp.raise_for_status()
|
|
except requests.exceptions.HTTPError:
|
|
return None
|
|
prefixes = resp.text.split("\n")
|
|
prefixes.pop(0)
|
|
return prefixes
|