From a09b3ec20a745cfd1bccb7bd9db513bfe36968db Mon Sep 17 00:00:00 2001
From: puuu <puuu@users.noreply.github.com>
Date: Wed, 21 Jun 2017 16:01:10 +0900
Subject: [PATCH] umqtt.simple: Do DNS lookup in connect() method.

---
 umqtt.simple/umqtt/simple.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/umqtt.simple/umqtt/simple.py b/umqtt.simple/umqtt/simple.py
index 8a0aa53d..8216fa5e 100644
--- a/umqtt.simple/umqtt/simple.py
+++ b/umqtt.simple/umqtt/simple.py
@@ -13,7 +13,8 @@ class MQTTClient:
             port = 8883 if ssl else 1883
         self.client_id = client_id
         self.sock = None
-        self.addr = socket.getaddrinfo(server, port)[0][-1]
+        self.server = server
+        self.port = port
         self.ssl = ssl
         self.ssl_params = ssl_params
         self.pid = 0
@@ -53,7 +54,8 @@ class MQTTClient:
 
     def connect(self, clean_session=True):
         self.sock = socket.socket()
-        self.sock.connect(self.addr)
+        addr = socket.getaddrinfo(self.server, self.port)[0][-1]
+        self.sock.connect(addr)
         if self.ssl:
             import ussl
             self.sock = ussl.wrap_socket(self.sock, **self.ssl_params)