From 557336791333ea8793301eefbc229d8fd88869c1 Mon Sep 17 00:00:00 2001 From: Shoghi Cervantes Date: Sat, 10 Jan 2015 00:24:43 +0100 Subject: [PATCH] Resolve DNS before sending packets --- mcstatus/server.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/mcstatus/server.py b/mcstatus/server.py index 6fe12eb..9cc6d10 100644 --- a/mcstatus/server.py +++ b/mcstatus/server.py @@ -62,9 +62,17 @@ class MinecraftServer: def query(self, retries=3): exception = None + host = self.host + try: + answers = dns.resolver.query(host, "A") + if len(answers): + answer = answers[0] + host = str(answer).rstrip(".") + except Exception as e: + pass for attempt in range(retries): try: - connection = UDPSocketConnection((self.host, self.port)) + connection = UDPSocketConnection((host, self.port)) querier = ServerQuerier(connection) querier.handshake() return querier.read_query()