umqtt.simple: Do DNS lookup in connect() method.

puuu · pfalcon · commit a09b3ec20a74 · 2017-06-27T21:16:13.000+03:00
diff --git a/umqtt.simple/umqtt/simple.py b/umqtt.simple/umqtt/simple.py
@@ -13,7 +13,8 @@ def __init__(self, client_id, server, port=0, user=None, password=None, keepaliv
             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 @@ def set_last_will(self, topic, msg, retain=False, qos=0):
 
     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)
