Added an enum-like class for the WiFi connection type and refactored the code

Author: docmollo

adafruit_esp32spi/adafruit_esp32spi_wifimanager.py

@@ -34,18 +34,31 @@
 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, con_type=1):
+    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 int con_type: (Optional) Type of WiFi connection to make: normal=1, WPA2 Enterprise=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
@@ -57,7 +70,7 @@ def __init__(self, esp, secrets, status_pixel=None, attempts=2, con_type=1):
         self.ent_user = secrets['ent_user']
         self.ent_passwd = secrets['ent_passwd']
         self.attempts = attempts
-        self.con_type = con_type
+        self.wificonntype = wificonntype
         requests.set_interface(self._esp)
         self.statuspix = status_pixel
         self.pixel_status(0)
@@ -82,7 +95,7 @@ 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
-        if self.con_type == 1:
+        if self.wificonntype == WiFiConnType.normal:
             while not self._esp.is_connected:
                 try:
                     if self.debug:
@@ -98,7 +111,7 @@ def connect(self):
                         failure_count = 0
                         self.reset()
                     continue
-        elif self.con_type == 2:
+        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'))