Examples and tests for debugging

adafruit_esp32spi/adafruit_esp32spi_socket.py

@@ -39,7 +39,7 @@ def getaddrinfo(host, port, family=0, socktype=0, proto=0, flags=0):
     """Given a hostname and a port name, return a 'socket.getaddrinfo'
     compatible list of tuples. Honestly, we ignore anything but host & port"""
     if not isinstance(port, int):
-        raise RuntimeError("Port must be an integer")
+        raise ValueError("Port must be an integer")
     ipaddr = _the_interface.get_host_by_name(host)
     return [(AF_INET, socktype, proto, "", (ipaddr, port))]
 
@@ -56,7 +56,7 @@ def __init__(
         self, family=AF_INET, type=SOCK_STREAM, proto=0, fileno=None, socknum=None
     ):
         if family != AF_INET:
-            raise RuntimeError("Only AF_INET family supported")
+            raise ValueError("Only AF_INET family supported")
         self._type = type
         self._buffer = b""
         self._socknum = socknum if socknum else _the_interface.get_socket()
@@ -74,7 +74,7 @@ def connect(self, address, conntype=None):
         if not _the_interface.socket_connect(
             self._socknum, host, port, conn_mode=conntype
         ):
-            raise RuntimeError("Failed to connect to host", host)
+            raise ConnectionError("Failed to connect to host", host)
         self._buffer = b""
 
     def send(self, data):  # pylint: disable=no-self-use
@@ -105,7 +105,7 @@ def readline(self, eol=b"\r\n"):
                 self._buffer += _the_interface.socket_read(self._socknum, avail)
             elif self._timeout > 0 and time.monotonic() - stamp > self._timeout:
                 self.close()  # Make sure to close socket so that we don't exhaust sockets.
-                raise RuntimeError("Didn't receive full response, failing out")
+                raise OSError("Didn't receive full response, failing out")
         firstline, self._buffer = self._buffer.split(eol, 1)
         gc.collect()
         return firstline

adafruit_esp32spi/adafruit_esp32spi_wifimanager.py

@@ -144,7 +144,7 @@ def connect_normal(self):
                 self.esp.connect_AP(bytes(ssid, "utf-8"), bytes(password, "utf-8"))
                 failure_count = 0
                 self.pixel_status((0, 100, 0))
-            except (ValueError, RuntimeError) as error:
+            except OSError as error:
                 print("Failed to connect, retrying\n", error)
                 failure_count += 1
                 if failure_count >= self.attempts:
@@ -173,7 +173,7 @@ def create_ap(self):
                     self.esp.create_AP(bytes(self.ssid, "utf-8"), None)
                 failure_count = 0
                 self.pixel_status((0, 100, 0))
-            except (ValueError, RuntimeError) as error:
+            except OSError as error:
                 print("Failed to create access point\n", error)
                 failure_count += 1
                 if failure_count >= self.attempts:
@@ -203,7 +203,7 @@ def connect_enterprise(self):
                 failure_count = 0
                 self.pixel_status((0, 100, 0))
                 sleep(1)
-            except (ValueError, RuntimeError) as error:
+            except OSError as error:
                 print("Failed to connect, retrying\n", error)
                 failure_count += 1
                 if failure_count >= self.attempts:

adafruit_esp32spi/adafruit_esp32spi_wsgiserver.py

@@ -107,7 +107,7 @@ def update_poll(self):
 
     def finish_response(self, result):
         """
-        Called after the application callbile returns result data to respond with.
+        Called after the application callable returns result data to respond with.
         Creates the HTTP Response payload from the response_headers and results data,
         and sends it back to client.
 
@@ -138,23 +138,21 @@ def client_available(self):
         :rtype: Socket
         """
         sock = None
-        if self._server_sock.socknum != NO_SOCK_AVAIL:
-            if self._client_sock.socknum != NO_SOCK_AVAIL:
-                # check previous received client socket
-                if self._debug > 2:
-                    print("checking if last client sock still valid")
-                if self._client_sock.connected() and self._client_sock.available():
-                    sock = self._client_sock
-            if not sock:
-                # check for new client sock
-                if self._debug > 2:
-                    print("checking for new client sock")
-                client_sock_num = _the_interface.socket_available(
-                    self._server_sock.socknum
-                )
-                sock = socket.socket(socknum=client_sock_num)
-        else:
-            print("Server has not been started, cannot check for clients!")
+        if self._server_sock.socknum == NO_SOCK_AVAIL:
+            raise ValueError("Server has not been started, cannot check for clients!")
+
+        if self._client_sock.socknum != NO_SOCK_AVAIL:
+            # check previous received client socket
+            if self._debug > 2:
+                print("checking if last client sock still valid")
+            if self._client_sock.connected() and self._client_sock.available():
+                sock = self._client_sock
+        if not sock:
+            # check for new client sock
+            if self._debug > 2:
+                print("checking for new client sock")
+            client_sock_num = _the_interface.socket_available(self._server_sock.socknum)
+            sock = socket.socket(socknum=client_sock_num)
 
         if sock and sock.socknum != NO_SOCK_AVAIL:
             if self._debug > 2:

adafruit_esp32spi/digitalio.py

@@ -61,7 +61,7 @@ def init(self, mode=IN):
                 self._mode = self.OUT
                 self._esp.set_pin_mode(self.pin_id, 1)
             else:
-                raise RuntimeError("Invalid mode defined")
+                raise ValueError("Invalid mode defined")
 
     def value(self, val=None):
         """Sets ESP32 Pin GPIO output mode.
@@ -76,7 +76,7 @@ def value(self, val=None):
                 self._value = val
                 self._esp.set_digital_write(self.pin_id, 1)
             else:
-                raise RuntimeError("Invalid value for pin")
+                raise ValueError("Invalid value for pin")
         else:
             raise NotImplementedError(
                 "digitalRead not currently implemented in esp32spi"

examples/esp32spi_aio_post.py

@@ -66,7 +66,7 @@
         response.close()
         counter = counter + 1
         print("OK")
-    except (ValueError, RuntimeError) as e:
+    except OSError as e:
         print("Failed to get data, retrying\n", e)
         wifi.reset()
         continue

examples/esp32spi_cheerlights.py

@@ -54,7 +54,7 @@
             value = value[key]
             print(value)
         response.close()
-    except (ValueError, RuntimeError) as e:
+    except OSError as e:
         print("Failed to get data, retrying\n", e)
         wifi.reset()
         continue

examples/esp32spi_get_board_details.py

@@ -0,0 +1,33 @@
+# SPDX-FileCopyrightText: 2020 ladyada for Adafruit Industries
+# SPDX-License-Identifier: MIT
+
+import time
+import board
+import busio
+
+from digitalio import DigitalInOut
+from adafruit_esp32spi import adafruit_esp32spi
+
+print("ESP32 SPI hardware test")
+
+# PyPortal or similar; edit pins as needed
+
+spi = busio.SPI(board.GP14, board.GP11, board.GP12)
+esp32_cs = DigitalInOut(board.GP10)
+esp32_ready = DigitalInOut(board.GP9)
+esp32_reset = DigitalInOut(board.GP8)
+esp = adafruit_esp32spi.ESP_SPIcontrol(
+    spi, esp32_cs, esp32_ready, esp32_reset, debug=True
+)
+
+# Fetch and print status
+if esp.status == adafruit_esp32spi.WL_IDLE_STATUS:
+    print("ESP32 found and in idle mode")
+print("Firmware version.", esp.firmware_version)
+print("MAC addr:", [hex(i) for i in esp.MAC_address])
+
+time.sleep(5)
+
+for ap in esp.scan_networks():
+    print("\t%s\t\tRSSI: %d" % (str(ap["ssid"], "utf-8"), ap["rssi"]))
+print("Done!")

examples/esp32spi_ipconfig.py

@@ -49,7 +49,7 @@
 while not esp.is_connected:
     try:
         esp.connect_AP(secrets["ssid"], secrets["password"])
-    except RuntimeError as e:
+    except OSError as e:
         print("could not connect to AP, retrying: ", e)
         continue
 print("Connected to", str(esp.ssid, "utf-8"), "\tRSSI:", esp.rssi)

examples/esp32spi_localtime.py

@@ -42,7 +42,7 @@
         print("Fetching json from", TIME_API)
         response = wifi.get(TIME_API)
         break
-    except (ValueError, RuntimeError) as e:
+    except OSError as e:
         print("Failed to get data, retrying\n", e)
         continue
 

examples/esp32spi_simpletest.py

@@ -59,7 +59,7 @@
 while not esp.is_connected:
     try:
         esp.connect_AP(secrets["ssid"], secrets["password"])
-    except RuntimeError as e:
+    except OSError as e:
         print("could not connect to AP, retrying: ", e)
         continue
 print("Connected to", str(esp.ssid, "utf-8"), "\tRSSI:", esp.rssi)

examples/esp32spi_simpletest_rp2040.py

@@ -42,7 +42,7 @@
 while not esp.is_connected:
     try:
         esp.connect_AP(secrets["ssid"], secrets["password"])
-    except RuntimeError as e:
+    except OSError as e:
         print("could not connect to AP, retrying: ", e)
         continue
 print("Connected to", str(esp.ssid, "utf-8"), "\tRSSI:", esp.rssi)

examples/esp32spi_wpa2ent_aio_post.py

@@ -63,7 +63,7 @@
         response.close()
         counter = counter + 1
         print("OK")
-    except (ValueError, RuntimeError) as e:
+    except OSError as e:
         print("Failed to get data, retrying\n", e)
         wifi.reset()
         continue

examples/gpio/esp32spi_gpio.py

@@ -94,7 +94,7 @@ def esp_init_pin_modes(din, dout):
         esp_init_pin_modes(ESP_D_R_PIN, ESP_D_W_PIN)
         esp_d_r_val = esp.set_digital_read(ESP_D_R_PIN)
         print("--> ESP read:", esp_d_r_val)
-    except (RuntimeError, AssertionError) as e:
+    except OSError as e:
         print("ESP32 Error", e)
         esp_reset_all()
 
@@ -104,7 +104,7 @@ def esp_init_pin_modes(din, dout):
         esp_init_pin_modes(ESP_D_R_PIN, ESP_D_W_PIN)
         esp.set_digital_write(ESP_D_W_PIN, esp_d_w_val)
         print("ESP wrote:", esp_d_w_val, "--> Red LED")
-    except (RuntimeError) as e:
+    except OSError as e:
         print("ESP32 Error", e)
         esp_reset_all()
 
@@ -121,7 +121,7 @@ def esp_init_pin_modes(din, dout):
             "v)",
             sep="",
         )
-    except (RuntimeError, AssertionError) as e:
+    except OSError as e:
         print("ESP32 Error", e)
         esp_reset_all()
 
@@ -166,7 +166,7 @@ def esp_init_pin_modes(din, dout):
             sep="",
         )
 
-    except (RuntimeError) as e:
+    except OSError as e:
         print("ESP32 Error", e)
         esp_reset_all()
 

examples/server/esp32spi_wsgiserver.py

@@ -213,7 +213,7 @@ def led_color(environ):  # pylint: disable=unused-argument
                 static
             )
         )
-except (OSError) as e:
+except OSError as e:
     raise RuntimeError(
         """
         This example depends on a static asset directory.
@@ -240,7 +240,7 @@ def led_color(environ):  # pylint: disable=unused-argument
     try:
         wsgiServer.update_poll()
         # Could do any other background tasks here, like reading sensors
-    except (ValueError, RuntimeError) as e:
+    except OSError as e:
         print("Failed to update server, restarting ESP32\n", e)
         wifi.reset()
         continue

examples/stress_tests/socket_client_server_stress/esp_rp2040_client/cp_socket_count_client.py

@@ -0,0 +1,55 @@
+# SPDX-FileCopyrightText: 2019 ladyada for Adafruit Industries
+# SPDX-License-Identifier: MIT
+
+import time
+import struct
+import board
+import busio
+from digitalio import DigitalInOut
+
+from adafruit_esp32spi import adafruit_esp32spi
+import adafruit_esp32spi.adafruit_esp32spi_socket as socket
+
+# Get wifi details and more from a secrets.py file
+try:
+    from secrets import secrets
+except ImportError:
+    print("WiFi secrets are kept in secrets.py, please add them there!")
+    raise
+
+print("ESP32 SPI socket count client test")
+
+TIMEOUT = 5
+# edit host and port to match server
+HOST = "192.168.1.149"
+PORT = 8981
+
+esp32_cs = DigitalInOut(board.GP10)
+esp32_ready = DigitalInOut(board.GP9)
+esp32_reset = DigitalInOut(board.GP8)
+spi = busio.SPI(board.GP14, board.GP11, board.GP12)
+esp = adafruit_esp32spi.ESP_SPIcontrol(spi, esp32_cs, esp32_ready, esp32_reset)
+
+# connect to wifi AP
+esp.connect(secrets)
+
+# test for connectivity to server
+print("Server ping:", esp.ping(HOST), "ms")
+
+# create the socket
+socket.set_interface(esp)
+socketaddr = socket.getaddrinfo(HOST, PORT)[0][4]
+s = socket.socket()
+s.settimeout(TIMEOUT)
+
+
+while True:
+    print("Connecting")
+    s.connect(socketaddr)
+    # get a count from the Socket. lets receive this as 4 bytes - unpack as an int.
+    data = s.recv(4)
+    # print it
+    print(f"Data length {len(data)}. Data: ", end="")
+    print(struct.unpack("!I", data))
+    time.sleep(0.01)
+    s.close()