Merge pull request #26 from shoghicp/master

Resolve DNS before sending packets
This commit is contained in:
Dinner Bone
2015-01-15 16:55:22 +01:00

View File

@@ -62,9 +62,17 @@ class MinecraftServer:
def query(self, retries=3): def query(self, retries=3):
exception = None 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): for attempt in range(retries):
try: try:
connection = UDPSocketConnection((self.host, self.port)) connection = UDPSocketConnection((host, self.port))
querier = ServerQuerier(connection) querier = ServerQuerier(connection)
querier.handshake() querier.handshake()
return querier.read_query() return querier.read_query()