refactor, remove debug output
This commit is contained in:
		
							
								
								
									
										14
									
								
								Makefile
									
									
									
									
									
								
							
							
						
						
									
										14
									
								
								Makefile
									
									
									
									
									
								
							@@ -1,18 +1,22 @@
 | 
				
			|||||||
SERIAL_PORT=/dev/tty.SLAB_USBtoUART
 | 
					#SERIAL_PORT=/dev/tty.SLAB_USBtoUART
 | 
				
			||||||
#SERIAL_PORT=/dev/tty.wchusbserial1420
 | 
					SERIAL_PORT=/dev/tty.wchusbserial1410
 | 
				
			||||||
#SERIAL_PORT=/dev/ttyUSB0
 | 
					#SERIAL_PORT=/dev/ttyUSB0
 | 
				
			||||||
 | 
					
 | 
				
			||||||
all: flash update config
 | 
					all: flash lib config update reboot
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					reboot:
 | 
				
			||||||
 | 
						esptool.py -p $(SERIAL_PORT) --after hard_reset read_mac
 | 
				
			||||||
 | 
					
 | 
				
			||||||
 | 
					lib:
 | 
				
			||||||
 | 
						sleep 1
 | 
				
			||||||
	ampy -p $(SERIAL_PORT) put uwebsockets.py
 | 
						ampy -p $(SERIAL_PORT) put uwebsockets.py
 | 
				
			||||||
 | 
					
 | 
				
			||||||
update:
 | 
					update:
 | 
				
			||||||
	sleep 3
 | 
					 | 
				
			||||||
	ampy -p $(SERIAL_PORT) put hal.py
 | 
						ampy -p $(SERIAL_PORT) put hal.py
 | 
				
			||||||
	ampy -p $(SERIAL_PORT) put main.py
 | 
						ampy -p $(SERIAL_PORT) put main.py
 | 
				
			||||||
	ampy -p $(SERIAL_PORT) put boot.py
 | 
						ampy -p $(SERIAL_PORT) put boot.py
 | 
				
			||||||
 | 
					
 | 
				
			||||||
config:
 | 
					config:
 | 
				
			||||||
	sleep 3
 | 
					 | 
				
			||||||
	ampy -p $(SERIAL_PORT) put config.json
 | 
						ampy -p $(SERIAL_PORT) put config.json
 | 
				
			||||||
 | 
					
 | 
				
			||||||
flash:
 | 
					flash:
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										4
									
								
								boot.py
									
									
									
									
									
								
							
							
						
						
									
										4
									
								
								boot.py
									
									
									
									
									
								
							@@ -3,7 +3,7 @@ import network
 | 
				
			|||||||
from hal import *
 | 
					from hal import *
 | 
				
			||||||
from time import sleep
 | 
					from time import sleep
 | 
				
			||||||
 | 
					
 | 
				
			||||||
print("Press Ctrl-C to stop boot script...")
 | 
					#print("Press Ctrl-C to stop boot script...")
 | 
				
			||||||
sleep(0.2)
 | 
					sleep(0.2)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# open and parse the config file
 | 
					# open and parse the config file
 | 
				
			||||||
@@ -25,7 +25,7 @@ if not wlan.isconnected():
 | 
				
			|||||||
        ssid = network[0].decode("utf-8")
 | 
					        ssid = network[0].decode("utf-8")
 | 
				
			||||||
        # check if the SSID is in the config file
 | 
					        # check if the SSID is in the config file
 | 
				
			||||||
        if ssid in config["wifis"].keys():
 | 
					        if ssid in config["wifis"].keys():
 | 
				
			||||||
            print("connecting to: " + ssid)
 | 
					            #print("connecting to: " + ssid)
 | 
				
			||||||
            # start to connect to the pre-configured network
 | 
					            # start to connect to the pre-configured network
 | 
				
			||||||
            wlan.connect(ssid, config["wifis"][ssid])
 | 
					            wlan.connect(ssid, config["wifis"][ssid])
 | 
				
			||||||
            break
 | 
					            break
 | 
				
			||||||
 
 | 
				
			|||||||
							
								
								
									
										23
									
								
								main.py
									
									
									
									
									
								
							
							
						
						
									
										23
									
								
								main.py
									
									
									
									
									
								
							@@ -2,12 +2,12 @@ import _thread
 | 
				
			|||||||
import ubinascii
 | 
					import ubinascii
 | 
				
			||||||
import uwebsockets
 | 
					import uwebsockets
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# to remember WiFi disconnects
 | 
					 | 
				
			||||||
has_wifi_connection = False
 | 
					 | 
				
			||||||
# WebSocket connection
 | 
					# WebSocket connection
 | 
				
			||||||
conn = None
 | 
					conn = None
 | 
				
			||||||
# blockly highlihgt WebSocket connection
 | 
					# blockly highlihgt WebSocket connection
 | 
				
			||||||
conn_highlight = None
 | 
					conn_highlight = None
 | 
				
			||||||
 | 
					# to remember WiFi disconnects
 | 
				
			||||||
 | 
					has_wifi_connection = False
 | 
				
			||||||
# extract a unique name for the robot from the device MAC address
 | 
					# extract a unique name for the robot from the device MAC address
 | 
				
			||||||
name = "sumo-%s" % ubinascii.hexlify(wlan.config("mac")[-3:]).decode("ascii")
 | 
					name = "sumo-%s" % ubinascii.hexlify(wlan.config("mac")[-3:]).decode("ascii")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -73,12 +73,10 @@ def ws_handler():
 | 
				
			|||||||
            # continue to wait for a WiFi connection
 | 
					            # continue to wait for a WiFi connection
 | 
				
			||||||
            continue
 | 
					            continue
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        try:
 | 
					        try: # try to read from the WebSocket
 | 
				
			||||||
            fin, opcode, data = conn.read_frame()
 | 
					            fin, opcode, data = conn.read_frame()
 | 
				
			||||||
        except: # urror
 | 
					        except: # socket timeout, no data received
 | 
				
			||||||
            # socket timeout, no data received
 | 
					            # continue to reconnect to WiFi
 | 
				
			||||||
            print("socket timeout")
 | 
					 | 
				
			||||||
            # continue to read again from socket
 | 
					 | 
				
			||||||
            continue
 | 
					            continue
 | 
				
			||||||
 | 
					
 | 
				
			||||||
        if data == b"forward":
 | 
					        if data == b"forward":
 | 
				
			||||||
@@ -114,24 +112,25 @@ def ws_handler():
 | 
				
			|||||||
            sumorobot.calibrate_line()
 | 
					            sumorobot.calibrate_line()
 | 
				
			||||||
            #print("Got calibrate")
 | 
					            #print("Got calibrate")
 | 
				
			||||||
        elif b"Gone" in data:
 | 
					        elif b"Gone" in data:
 | 
				
			||||||
            print("Server said 410 Gone, attempting to reconnect...")
 | 
					            #print("Server said 410 Gone, attempting to reconnect...")
 | 
				
			||||||
            conn = uwebsockets.connect(url)
 | 
					            conn = uwebsockets.connect(url)
 | 
				
			||||||
        else:
 | 
					        else:
 | 
				
			||||||
            print("unknown command:", data)
 | 
					            pass
 | 
				
			||||||
 | 
					            #print("unknown command:", data)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# wait for WiFi to get connected
 | 
					# wait for WiFi to get connected
 | 
				
			||||||
while not wlan.isconnected():
 | 
					while not wlan.isconnected():
 | 
				
			||||||
    sleep_ms(100)
 | 
					    sleep_ms(100)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# connect to the websocket
 | 
					# connect to the websocket
 | 
				
			||||||
print("Connecting to:", url)
 | 
					#print("Connecting to:", url)
 | 
				
			||||||
conn = uwebsockets.connect(url)
 | 
					conn = uwebsockets.connect(url)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# set X seconds timeout for socket reads
 | 
					# set X seconds timeout for socket reads
 | 
				
			||||||
conn.settimeout(1)
 | 
					conn.settimeout(1)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
# send a ping to the robot
 | 
					# send a ping to the robot
 | 
				
			||||||
print("Sending ping")
 | 
					#print("Sending ping")
 | 
				
			||||||
conn.send("{'ping': true}")
 | 
					conn.send("{'ping': true}")
 | 
				
			||||||
conn.send("{'ip': '" + wlan.ifconfig()[0] + "'}")
 | 
					conn.send("{'ip': '" + wlan.ifconfig()[0] + "'}")
 | 
				
			||||||
 | 
					
 | 
				
			||||||
@@ -144,7 +143,7 @@ sumorobot = Sumorobot(conn_highlight.send)
 | 
				
			|||||||
# indicate that the WebSocket is connected
 | 
					# indicate that the WebSocket is connected
 | 
				
			||||||
sumorobot.set_led(STATUS, True)
 | 
					sumorobot.set_led(STATUS, True)
 | 
				
			||||||
 | 
					
 | 
				
			||||||
print("Starting WebSocket and code loop")
 | 
					#print("Starting WebSocket and code loop")
 | 
				
			||||||
# start the code processing thread
 | 
					# start the code processing thread
 | 
				
			||||||
_thread.start_new_thread(step, ())
 | 
					_thread.start_new_thread(step, ())
 | 
				
			||||||
# start the Websocket processing thread
 | 
					# start the Websocket processing thread
 | 
				
			||||||
 
 | 
				
			|||||||
		Reference in New Issue
	
	Block a user