repos/pypilot
4
0
Fork 0
Code Issues Pull Requests Actions Packages Releases 11 Activity

Fix Sonar issue, formatting (#99)

Browse Source 
* Fix Sonar issue, formatting

* Remove dev dependencies from requirements.tzt

* Fix unit test failure
This commit is contained in:
Luke Tainton
2023-06-04 12:06:30 +01:00
committed by GitHub
parent dd8411d726
commit 950d1164eb
10 changed files with 48 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app/ip_info.py
View File

@ -2,19 +2,18 @@
"""MODULE: Provides functions to call various APIs to retrieve IP/prefix information."""
from typing import Union
import ipaddress
import requests
def get_ip_information(ipv4_address: ipaddress.IPv4Address) -> dict:
def get_ip_information(ipv4_address: ipaddress.IPv4Address) -> dict | None:
"""Retrieves information about a given IPv4 address from IP-API.com."""
api_endpoint = f"http://ip-api.com/json/{ipv4_address}"
api_endpoint: str = f"http://ip-api.com/json/{ipv4_address}"
try:
resp = requests.get(api_endpoint, timeout=10)
resp: requests.Response = requests.get(api_endpoint, timeout=10)
resp.raise_for_status()
ret = resp.json() if resp.json().get("status") == "success" else None
ret: dict | None = resp.json() if resp.json().get("status") == "success" else None
except (requests.exceptions.JSONDecodeError, requests.exceptions.HTTPError):
ret = None
return ret
@ -22,18 +21,18 @@ def get_ip_information(ipv4_address: ipaddress.IPv4Address) -> dict:
def get_autonomous_system_number(as_info: str) -> str:
"""Parses AS number from provided AS information."""
as_number = as_info.split(" ")[0]
as_number: str = as_info.split(" ")[0]
return as_number
def get_prefix_information(autonomous_system: int) -> Union[list, None]:
def get_prefix_information(autonomous_system: str) -> list | None:
"""Retrieves prefix information about a given autonomous system."""
api_endpoint = f"https://api.hackertarget.com/aslookup/?q={str(autonomous_system)}"
api_endpoint: str = f"https://api.hackertarget.com/aslookup/?q={str(autonomous_system)}"
try:
resp = requests.get(api_endpoint, timeout=10)
resp: requests.Response = requests.get(api_endpoint, timeout=10)
resp.raise_for_status()
except requests.exceptions.HTTPError:
return None
prefixes = resp.text.split("\n")
prefixes: list[str] = resp.text.split("\n")
prefixes.pop(0)
return prefixes