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

Fix unit test failure

Browse Source
This commit is contained in:
Luke Tainton
2023-06-04 12:02:26 +01:00
parent a2455dff88
commit 7f2766757c
3 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/query_normalisation.py
View File

@@ -21,10 +21,11 @@ def is_ip_address(query: str) -> bool:
def resolve_domain_name(domain_name: str) -> ipaddress.IPv4Address | None:
"""Resolve a domain name via DNS or return None."""
try:
ip_address: ipaddress.IPv4Address = ipaddress.IPv4Address(socket.gethostbyname(domain_name))
except socket.gaierror:
ip_address = None
return ip_address
result: str = socket.gethostbyname(domain_name)
ip_address: ipaddress.IPv4Address = ipaddress.IPv4Address(result)
return ip_address
except (socket.gaierror, ipaddress.AddressValueError):
return None
def get_public_ip() -> ipaddress.IPv4Address: