umqtt.simple: Be sure to return socket to blocking state after check_msg().

Author: pfalcon

umqtt.simple/umqtt/simple.py

@@ -112,9 +112,9 @@ def subscribe(self, topic, qos=0):
     # messages processed internally.
     def wait_msg(self):
         res = self.sock.read(1)
+        self.sock.setblocking(True)
         if res is None:
             return None
-        self.sock.setblocking(True)
         if res == b"":
             raise OSError(-1)
         if res == b"\xd0":  # PINGRESP