Skip to content

Commit 3acedda

Browse files
authored
Merge pull request adafruit#8 from brentru/switch-read
Use socket.recv
2 parents 5c953e2 + 9f08371 commit 3acedda

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

adafruit_minimqtt.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,7 @@ def connect(self, clean_session=True):
275275
while True:
276276
op = self._wait_for_msg()
277277
if op == 32:
278-
rc = self._sock.read(3)
278+
rc = self._sock.recv(3)
279279
assert rc[0] == 0x02
280280
if rc[2] != 0x00:
281281
raise MMQTTException(CONNACK_ERRORS[rc[2]])
@@ -313,7 +313,7 @@ def ping(self):
313313
while True:
314314
op = self._wait_for_msg(0.5)
315315
if op == 208:
316-
ping_resp = self._sock.read(2)
316+
ping_resp = self._sock.recv(2)
317317
if ping_resp[0] != 0x00:
318318
raise MMQTTException('PINGRESP not returned from broker.')
319319
return
@@ -394,9 +394,9 @@ def publish(self, topic, msg, retain=False, qos=0):
394394
while True:
395395
op = self._wait_for_msg()
396396
if op == 0x40:
397-
sz = self._sock.read(1)
397+
sz = self._sock.recv(1)
398398
assert sz == b"\x02"
399-
rcv_pid = self._sock.read(2)
399+
rcv_pid = self._sock.recv(2)
400400
rcv_pid = rcv_pid[0] << 0x08 | rcv_pid[1]
401401
if pid == rcv_pid:
402402
if self.on_publish is not None:
@@ -472,7 +472,7 @@ def subscribe(self, topic, qos=0):
472472
while True:
473473
op = self._wait_for_msg()
474474
if op == 0x90:
475-
rc = self._sock.read(4)
475+
rc = self._sock.recv(4)
476476
assert rc[1] == packet[2] and rc[2] == packet[3]
477477
if rc[3] == 0x80:
478478
raise MMQTTException('SUBACK Failure!')
@@ -529,7 +529,7 @@ def unsubscribe(self, topic):
529529
while True:
530530
op = self._wait_for_msg()
531531
if op == 176:
532-
return_code = self._sock.read(3)
532+
return_code = self._sock.recv(3)
533533
assert return_code[0] == 0x02
534534
# [MQTT-3.32]
535535
assert return_code[1] == packet_id_bytes[0] and return_code[2] == packet_id_bytes[1]
@@ -640,27 +640,27 @@ def _wait_for_msg(self, timeout=30):
640640
"""Reads and processes network events.
641641
Returns response code if successful.
642642
"""
643-
res = self._sock.read(1)
643+
res = self._sock.recv(1)
644644
self._sock.settimeout(timeout)
645645
if res in [None, b""]:
646646
return None
647647
if res == MQTT_PINGRESP:
648-
sz = self._sock.read(1)[0]
648+
sz = self._sock.recv(1)[0]
649649
assert sz == 0
650650
return None
651651
if res[0] & 0xf0 != 0x30:
652652
return res[0]
653653
sz = self._recv_len()
654-
topic_len = self._sock.read(2)
654+
topic_len = self._sock.recv(2)
655655
topic_len = (topic_len[0] << 8) | topic_len[1]
656-
topic = self._sock.read(topic_len)
656+
topic = self._sock.recv(topic_len)
657657
topic = str(topic, 'utf-8')
658658
sz -= topic_len + 2
659659
if res[0] & 0x06:
660-
pid = self._sock.read(2)
660+
pid = self._sock.recv(2)
661661
pid = pid[0] << 0x08 | pid[1]
662662
sz -= 0x02
663-
msg = self._sock.read(sz)
663+
msg = self._sock.recv(sz)
664664
if self.on_message is not None:
665665
self.on_message(self, topic, str(msg, 'utf-8'))
666666
if res[0] & 0x06 == 0x02:
@@ -675,7 +675,7 @@ def _recv_len(self):
675675
n = 0
676676
sh = 0
677677
while True:
678-
b = self._sock.read(1)[0]
678+
b = self._sock.recv(1)[0]
679679
n |= (b & 0x7f) << sh
680680
if not b & 0x80:
681681
return n

0 commit comments

Comments
 (0)