60
app/ip_info.py
Normal file
60
app/ip_info.py
Normal file
@ -0,0 +1,60 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
"""MODULE: Provides functions to call various APIs to retrieve IP/prefix information."""
|
||||
|
||||
from typing import Optional
|
||||
|
||||
import ipaddress
|
||||
import requests
|
||||
|
||||
|
||||
def get_ip_information(ipv4_address: ipaddress.IPv4Address) -> Optional[dict]:
|
||||
"""Retrieves information about a given IPv4 address from IP-API.com.
|
||||
|
||||
Args:
|
||||
ipv4_address (ipaddress.IPv4Address): IPv4 address to query
|
||||
|
||||
Returns:
|
||||
Optional[dict]: API response
|
||||
"""
|
||||
api_endpoint: str = f"http://ip-api.com/json/{ipv4_address}"
|
||||
try:
|
||||
resp: requests.Response = requests.get(api_endpoint, timeout=10)
|
||||
resp.raise_for_status()
|
||||
ret: dict | None = 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.
|
||||
|
||||
Args:
|
||||
as_info (str): AS information
|
||||
|
||||
Returns:
|
||||
str: AS number
|
||||
"""
|
||||
as_number: str = as_info.split(" ")[0]
|
||||
return as_number
|
||||
|
||||
|
||||
def get_prefix_information(autonomous_system: str) -> Optional[list]:
|
||||
"""Retrieves prefix information about a given autonomous system.
|
||||
|
||||
Args:
|
||||
autonomous_system (str): autonomous system to query, e.g. AS123
|
||||
|
||||
Returns:
|
||||
Optional[list]: API response
|
||||
"""
|
||||
api_endpoint: str = f"https://api.hackertarget.com/aslookup/?q={str(autonomous_system)}"
|
||||
try:
|
||||
resp: requests.Response = requests.get(api_endpoint, timeout=10)
|
||||
resp.raise_for_status()
|
||||
except requests.exceptions.HTTPError:
|
||||
return None
|
||||
prefixes: list[str] = resp.text.split("\n")
|
||||
prefixes.pop(0)
|
||||
return prefixes
|
Reference in New Issue
Block a user