apply black and add example

Author: Neradoc

adafruit_esp32spi/adafruit_esp32spi.py

@@ -410,21 +410,35 @@ def scan_networks(self):
 
     def set_ip_config(self, new_ip, new_gw, new_mask="255.255.255.0"):
         """Tells the ESP32 to set ip, gateway and network mask b"\xFF" """
-        resp = self._send_command_get_response(_SET_IP_CONFIG,
-                                               params= [b"\x00",self.unpretty_ip(new_ip),self.unpretty_ip(new_gw), self.unpretty_ip(new_mask)],
-                                               sent_param_len_16=False)
+        resp = self._send_command_get_response(
+            _SET_IP_CONFIG,
+            params=[
+                b"\x00",
+                self.unpretty_ip(new_ip),
+                self.unpretty_ip(new_gw),
+                self.unpretty_ip(new_mask),
+            ],
+            sent_param_len_16=False,
+        )
         return resp
 
     def set_dns_config(self, dns1, dns2="8.8.8.8"):
         """Tells the ESP32 to set DNS, with dns2 default to google dns=8.8.8.8"""
-        resp = self._send_command_get_response(_SET_DNS_CONFIG, [b"\x00", self.unpretty_ip(dns1), self.unpretty_ip(dns2)])
+        resp = self._send_command_get_response(
+            _SET_DNS_CONFIG, [b"\x00", self.unpretty_ip(dns1), self.unpretty_ip(dns2)]
+        )
         if resp[0][0] != 1:
             raise RuntimeError("Failed to set dns with esp32")
-        
+
     def set_hostname(self, hostname):
-        """Tells the ESP32 to set hostname"""
-        resp = self._send_command_get_response(_SET_HOSTNAME, [hostname])
-        return resp
+        """
+        Tells the ESP32 to set hostname
+
+        :params str hostname: Set the host name, used by DHCP to associate a local
+                              domain name like hostname.home for example, depending
+                              on the DHCP server setup.
+        """
+        resp = self._send_command_get_response(_SET_HOSTNAME, [hostname.encode()])
         if resp[0][0] != 1:
             raise RuntimeError("Failed to set hostname with esp32")
 

examples/esp32spi_ipconfig.py

@@ -0,0 +1,82 @@
+import time
+import board
+import busio
+from digitalio import DigitalInOut
+import adafruit_requests as requests
+import adafruit_esp32spi.adafruit_esp32spi_socket as socket
+from adafruit_esp32spi import adafruit_esp32spi
+import neopixel
+from rainbowio import colorwheel
+import gc
+from secrets import secrets
+
+IP_ADDRESS = "192.168.2.2"
+GATEWAY_ADDRESS = "192.168.2.1"
+SUBNET_MASK = "255.255.255.0"
+
+UDP_IN_ADDR = "192.168.2.2"
+UDP_IN_PORT = 5500
+
+UDP_TIMEOUT = 20
+
+esp32_cs = DigitalInOut(board.CS1)
+esp32_ready = DigitalInOut(board.ESP_BUSY)
+esp32_reset = DigitalInOut(board.ESP_RESET)
+
+spi = busio.SPI(board.SCK1, board.MOSI1, board.MISO1)
+
+esp = adafruit_esp32spi.ESP_SPIcontrol(spi, esp32_cs, esp32_ready, esp32_reset)
+socket.set_interface(esp)
+
+s_in = socket.socket(type=socket.SOCK_DGRAM)
+s_in.settimeout(UDP_TIMEOUT)
+print("set hostname:")
+esp.set_hostname("new_hostname".encode("utf-8"))
+
+if esp.status == adafruit_esp32spi.WL_IDLE_STATUS:
+    print("ESP32 found and in idle mode")
+print("Firmware vers.", esp.firmware_version)
+print("MAC addr:", [hex(i) for i in esp.MAC_address])
+
+print("Connecting to AP...")
+while not esp.is_connected:
+    try:
+        esp.connect_AP(secrets["ssid"], secrets["password"])
+    except RuntimeError as e:
+        print("could not connect to AP, retrying: ", e)
+        continue
+print("Connected to", str(esp.ssid, "utf-8"), "\tRSSI:", esp.rssi)
+ip1 = esp.ip_address
+
+print("set ip dns")
+esp.set_dns_config("192.168.2.1", "8.8.8.8")
+
+print("set ip config")
+esp.set_ip_config(IP_ADDRESS, GATEWAY_ADDRESS, SUBNET_MASK)
+
+time.sleep(1)
+ip2 = esp.ip_address
+
+time.sleep(1)
+info = esp.network_data
+print(
+    "get network_data: ",
+    esp.pretty_ip(info["ip_addr"]),
+    esp.pretty_ip(info["gateway"]),
+    esp.pretty_ip(info["netmask"]),
+)
+
+IP_ADDR = esp.pretty_ip(esp.ip_address)
+print("ip:", IP_ADDR)
+print("My IP address is", esp.pretty_ip(esp.ip_address))
+print("udp in addr: ", UDP_IN_ADDR, UDP_IN_PORT)
+
+socketaddr_udp_in = socket.getaddrinfo(UDP_IN_ADDR, UDP_IN_PORT)[0][4]
+s_in.connect(socketaddr_udp_in, conntype=esp.UDP_MODE)
+print("connected local UDP")
+
+while True:
+    data = s_in.recv(1205)
+    if len(data) >= 1:
+        data = data.decode("utf-8")
+        print(len(data), data)