diff --git a/tools/lib/filereader.py b/tools/lib/filereader.py index e9b8b4b2ce..3bfdc75feb 100644 --- a/tools/lib/filereader.py +++ b/tools/lib/filereader.py @@ -10,10 +10,11 @@ DATA_ENDPOINT = os.getenv("DATA_ENDPOINT", "http://data-raw.comma.internal/") def internal_source_available(): try: hostname = urlparse(DATA_ENDPOINT).hostname - if hostname: - socket.gethostbyname(hostname) - return True - except socket.gaierror: + port = urlparse(DATA_ENDPOINT).port or 80 + with socket.socket(socket.AF_INET,socket.SOCK_STREAM) as s: + s.connect((hostname, port)) + return True + except (socket.gaierror, ConnectionRefusedError): pass return False