MicroPython Forum (Archive)

Code: Select all

#  ESPNow NODES

import network
import utime
from esp import espnow

class NODE():
    def __init__(self):
        self.MAC  = None
        self.data = None
        

class ESP_NET():
    #              owner_node  critical_nodes  recommended_nodes
    def __init__(self, o_node, c_nodes = None, r_nodes = None ):
        print('Owner         :',o_node)
        print('Critical_nodes:',c_nodes)
        print('Recomend_nodes:',r_nodes)
        print()
        
        self.o_node  = o_node  # owner_node
        self.c_nodes = c_nodes # critical_nodes
        self.r_nodes = r_nodes # recommended_nodes
        
        #  A WLAN interface must be active to send()/recv()
        w0 = network.WLAN(network.STA_IF)
        w0.active(True)
        local = w0.config('mac')
        self.bcast = b'\xff'*6

        self.e0 = espnow.ESPNow()
        self.e0.init()
        self.e0.config(timeout=1000)

        self.e0.add_peer(local)      # add owner MAC first
        self.e0.add_peer(self.bcast) # add broadcast second
        
        self.macs = dict.fromkeys(c_nodes,None) # create NODE register
        
        find_peers = (len(c_nodes) if c_nodes else 0) + (len(r_nodes) if r_nodes else 0)
        i=0
        while True:
            ping = 'Ping'+o_node+' '+str(i)
            for mac, msg in self.e0:
                self.e0.send(self.bcast,ping,True) # I'm here
                print('Send: ',ping)
                    
                if mac:                            # Oy, who is here
                    print('MAC detected: ',mac)
                    new_msg = msg.decode('utf-8')
                    node_name = new_msg[4:5]
                    
                    if node_name in self.macs.keys(): # One of the declared Nodes
                        if self.macs[node_name]:
                            print('_____________Already peered with MAC: ',mac,'stored in ',self.macs)
                            if new_msg[0:4] == 'Pong':
                                self.macs[node_name].data = new_msg
                                print('_____________Pong from ',node_name)
                                find_peers -= 1
                        else: # New peer
                            self.e0.add_peer(mac)
                            new_MAC = self.e0.get_peer(mac)
                            self.macs[node_name] = NODE()
                            self.macs[node_name].MAC = new_MAC[0]
                            print('______________New peer with MAC: ',new_MAC[0])
                            self.e0.send(mac, bytearray('Pong'+o_node+' '+str(i)), True) #  send back answer

                        print ('Received from :{}: "{} peers left {}"'.format(node_name, new_msg,find_peers))
                        print()
                        break

            if not find_peers: break
            print ('..............waiting for msg '+str(i))
            print()
            i += 1


#################################################################################### main                
en = ESP_NET('A',['B','C'])

print('RESULT')
for node in en.macs.keys():
    print (node,en.macs[node].MAC,en.macs[node].data)