Adding WPA2 Enterprise support to the WiFi manager library

adafruit_esp32spi/adafruit_esp32spi_wifimanager.py

@@ -34,24 +34,43 @@
 from adafruit_esp32spi import adafruit_esp32spi
 import adafruit_esp32spi.adafruit_esp32spi_requests as requests
 
+class WiFiConnType: # pylint: disable=too-few-public-methods
+    """An enum-like class representing the different types of WiFi connections
+    that can be made. The values can be referenced like ``WiFiConnType.normal``.
+    Possible values are
+    - ``ThermocoupleType.normal``
+    - ``ThermocoupleType.enterprise``
+    """
+    # pylint: disable=invalid-name
+    normal = 1
+    enterprise = 2
+
 class ESPSPI_WiFiManager:
     """
     A class to help manage the Wifi connection
     """
-    def __init__(self, esp, secrets, status_pixel=None, attempts=2):
+    def __init__(self, esp, secrets, status_pixel=None, attempts=2, wificonntype=WiFiConnType.normal):
         """
         :param ESP_SPIcontrol esp: The ESP object we are using
         :param dict secrets: The WiFi and Adafruit IO secrets dict (See examples)
         :param status_pixel: (Optional) The pixel device - A NeoPixel or DotStar (default=None)
         :type status_pixel: NeoPixel or DotStar
         :param int attempts: (Optional) Failed attempts before resetting the ESP32 (default=2)
+        :param const con_type: (Optional) Type of WiFi connection: normal=1, WPA2 Enterprise=2
+        :param ~adafruit_esp32spi_wifimanager.WiFiConnType wificonntype: The type of WiFi \
+        connection to make. The default is "normal".
         """
         # Read the settings
         self._esp = esp
         self.debug = False
         self.ssid = secrets['ssid']
         self.password = secrets['password']
+        self.ent_ssid = secrets['ent_ssid']
+        self.ent_ident = secrets['ent_ident']
+        self.ent_user = secrets['ent_user']
+        self.ent_passwd = secrets['ent_passwd']
         self.attempts = attempts
+        self.wificonntype = wificonntype
         requests.set_interface(self._esp)
         self.statuspix = status_pixel
         self.pixel_status(0)
@@ -76,21 +95,45 @@ def connect(self):
             for access_pt in self._esp.scan_networks():
                 print("\t%s\t\tRSSI: %d" % (str(access_pt['ssid'], 'utf-8'), access_pt['rssi']))
         failure_count = 0
-        while not self._esp.is_connected:
-            try:
-                if self.debug:
-                    print("Connecting to AP...")
-                self.pixel_status((100, 0, 0))
-                self._esp.connect_AP(bytes(self.ssid, 'utf-8'), bytes(self.password, 'utf-8'))
-                failure_count = 0
-                self.pixel_status((0, 100, 0))
-            except (ValueError, RuntimeError) as error:
-                print("Failed to connect, retrying\n", error)
-                failure_count += 1
-                if failure_count >= self.attempts:
+        if self.wificonntype == WiFiConnType.normal:
+            while not self._esp.is_connected:
+                try:
+                    if self.debug:
+                        print("Connecting to AP...")
+                    self.pixel_status((100, 0, 0))
+                    self._esp.connect_AP(bytes(self.ssid, 'utf-8'), bytes(self.password, 'utf-8'))
+                    failure_count = 0
+                    self.pixel_status((0, 100, 0))
+                except (ValueError, RuntimeError) as error:
+                    print("Failed to connect, retrying\n", error)
+                    failure_count += 1
+                    if failure_count >= self.attempts:
+                        failure_count = 0
+                        self.reset()
+                    continue
+        elif self.wificonntype == WiFiConnType.enterprise:
+            self._esp.wifi_set_network(bytes(self.ent_ssid, 'utf-8'))
+            self._esp.wifi_set_entidentity(bytes(self.ent_ident, 'utf-8'))
+            self._esp.wifi_set_entusername(bytes(self.ent_user, 'utf-8'))
+            self._esp.wifi_set_entpassword(bytes(self.ent_passwd, 'utf-8'))
+            self._esp.wifi_set_entenable()
+            while not self._esp.is_connected:
+                try:
+                    if self.debug:
+                        print("Connecting to WPA2 Enterprise AP...")
+                    self.pixel_status((100, 0, 0))
                     failure_count = 0
-                    self.reset()
-                continue
+                    self.pixel_status((0, 100, 0))
+                except (ValueError, RuntimeError) as error:
+                    print("Failed to connect, retrying\n", error)
+                    failure_count += 1
+                    if failure_count >= self.attempts:
+                        failure_count = 0
+                        self.reset()
+                    continue
+        else:
+            print("Invalid connection type! Exiting...")
+            exit(1)
 
     def get(self, url, **kw):
         """