From 03e129f2a83443fb904ee10855724c27985a39a7 Mon Sep 17 00:00:00 2001 From: Luke Tainton Date: Sat, 22 Oct 2022 00:03:42 +0100 Subject: [PATCH] SONAR: pylint:W3101 --- app/ip_info.py | 4 ++-- app/query_normalisation.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/app/ip_info.py b/app/ip_info.py index 8084c12..8a8e6e5 100644 --- a/app/ip_info.py +++ b/app/ip_info.py @@ -12,7 +12,7 @@ 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 = requests.get(api_endpoint, timeout=10) resp.raise_for_status() ret = resp.json() if resp.json().get("status") == "success" else None except (requests.exceptions.JSONDecodeError, requests.exceptions.HTTPError): @@ -30,7 +30,7 @@ def get_prefix_information(autonomous_system: int) -> Union[list, None]: """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 = requests.get(api_endpoint, timeout=10) resp.raise_for_status() except requests.exceptions.HTTPError: return None diff --git a/app/query_normalisation.py b/app/query_normalisation.py index 3aa06b2..fb6b783 100644 --- a/app/query_normalisation.py +++ b/app/query_normalisation.py @@ -28,5 +28,5 @@ def resolve_domain_name(domain_name: str) -> ipaddress.IPv4Address: def get_public_ip() -> ipaddress.IPv4Address: """Get the user's current public IPv4 address.""" - ip_address = requests.get("https://api.ipify.org").text + ip_address = requests.get("https://api.ipify.org", timeout=10).text return ip_address