forked from micropython/micropython
/
espnow.py
46 lines (32 loc) · 1.21 KB
/
espnow.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# espnow module for MicroPython on ESP32
# MIT license; Copyright (c) 2022 Glenn Moloney @glenn20
from micropython import const
from _espnow import *
EVENT_RECV_MSG = const(1)
class ESPNow(ESPNow):
# Static buffers for alloc free receipt of messages with ESPNow.irecv().
_data = [None, bytearray(MAX_DATA_LEN)]
_none_tuple = (None, None)
def __init__(self):
super().__init__()
def irecv(self, timeout=None):
n = self.recvinto(self._data, timeout)
return self._data if n else self._none_tuple
def recv(self, timeout=None):
n = self.recvinto(self._data, timeout)
return [bytes(x) for x in self._data] if n else self._none_tuple
def on_recv(self, recv_cb, arg=None):
super().on_recv(recv_cb, self if arg is None else arg)
def irq(self, recv_cb): # irq() api
def cb_wrapper(e):
recv_cb(EVENT_RECV_MSG, e.irecv(0))
super().on_recv(cb_wrapper, self)
def __iter__(self):
return self
def __next__(self):
return self.irecv() # Use alloc free irecv() method
# Backward compatibility with pre-release API
def init(self):
self.active(True)
def deinit(self):
self.active(False)