pull in AIO group mqtt

Author: brentru

Adafruit_IO_Air_Quality/code.py

@@ -2,10 +2,37 @@
 import board
 import busio
 from digitalio import DigitalInOut, Direction, Pull
+import adafruit_dotstar as dotstar
+from adafruit_esp32spi import adafruit_esp32spi
+from adafruit_esp32spi import adafruit_esp32spi_wifimanager
+import adafruit_esp32spi.adafruit_esp32spi_socket as socket
+import adafruit_minimqtt.adafruit_minimqtt as MQTT
+from adafruit_io.adafruit_io import IO_MQTT
+
 from simpleio import map_range
+
 import adafruit_pm25
 import adafruit_bme280
 
+### WiFi ###
+
+# 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
+
+# If you have an externally connected ESP32:
+esp32_cs = DigitalInOut(board.D13)
+esp32_reset = DigitalInOut(board.D12)
+esp32_ready = DigitalInOut(board.D11)
+
+spi = busio.SPI(board.SCK, board.MOSI, board.MISO)
+esp = adafruit_esp32spi.ESP_SPIcontrol(spi, esp32_cs, esp32_ready, esp32_reset)
+status_light = dotstar.DotStar(board.APA102_SCK, board.APA102_MOSI, 1, brightness=0.2)
+wifi = adafruit_esp32spi_wifimanager.ESPSPI_WiFiManager(esp, secrets, status_light)
+
 
 # Connect to a PM2.5 sensor over UART
 uart = busio.UART(board.TX, board.RX, baudrate=9600)
@@ -15,6 +42,23 @@
 i2c = busio.I2C(board.SCL, board.SDA)
 bme280 = adafruit_bme280.Adafruit_BME280_I2C(i2c)
 
+### MiniMQTT Callback Functions ###
+def connected(client):
+    # Connected function will be called when the client is connected to Adafruit IO.
+    # This is a good place to subscribe to feed changes.  The client parameter
+    # passed to this function is the Adafruit IO MQTT client so you can make
+    # calls against it easily.
+    print("Connected to Adafruit IO!")
+
+    # Subscribe to Group
+    io.subscribe(group_key=group_name)
+
+# pylint: disable=unused-argument
+def disconnected(client):
+    # Disconnected function will be called when the client disconnects.
+    print("Disconnected from Adafruit IO!")
+
+### Sensor Functions ###
 def calculate_aqi(pm_sensor_reading):
     """Returns a calculated air quality index (AQI)
     and category as a tuple.
@@ -28,30 +72,30 @@ def calculate_aqi(pm_sensor_reading):
     if 0.0 <= pm_sensor_reading <= 12.0:
         # AQI calculation using EPA breakpoints (Ilow-IHigh)
         aqi = map_range(int(pm_sensor_reading), 0, 12, 0, 50)
-        aqi_condition = "Good"
+        aqi_category = "Good"
     elif 12.1 <= pm_sensor_reading <= 35.4:
         aqi = map_range(int(pm_sensor_reading), 12, 35, 51, 100)
-        aqi_condition = "Moderate"
+        aqi_category = "Moderate"
     elif 35.5 <= pm_sensor_reading <= 55.4:
         aqi = map_range(int(pm_sensor_reading), 36, 55, 101, 150)
-        aqi_condition = "Unhealthy for Sensitive Groups"
+        aqi_category = "Unhealthy for Sensitive Groups"
     elif 55.5 <= pm_sensor_reading <= 150.4:
         aqi = map_range(int(pm_sensor_reading), 56, 150, 151, 200)
-        aqi_condition = "Unhealthy"
+        aqi_category = "Unhealthy"
     elif 150.5 <= pm_sensor_reading <= 250.4:
         aqi = map_range(int(pm_sensor_reading), 151, 250, 201, 300)
-        aqi_condition = "Very Unhealthy"
+        aqi_category = "Very Unhealthy"
     elif 250.5 <= pm_sensor_reading <= 350.4:
         aqi = map_range(int(pm_sensor_reading), 251, 350, 301, 400)
-        aqi_condition = "Hazardous"
+        aqi_category = "Hazardous"
     elif 350.5 <= pm_sensor_reading <= 500.4:
         aqi = map_range(int(pm_sensor_reading), 351, 500, 401, 500)
-        aqi_condition = "Hazardous"
+        aqi_category = "Hazardous"
     else:
         print("Invalid PM2.5 concentration")
         aqi = -1
-        aqi_condition = None
-    return aqi, aqi_condition
+        aqi_category = None
+    return aqi, aqi_category
 
 def sample_aq_sensor():
     """Samples PM2.5 sensor
@@ -93,16 +137,65 @@ def read_bme280(is_celsius=False):
         temperature = temperature * 1.8 + 32
     return temperature, humidity
 
+### CODE ###
+
+# Connect to WiFi
+print("Connecting to WiFi...")
+wifi.connect()
+print("Connected!")
+
+# Initialize MQTT interface with the esp interface
+MQTT.set_socket(socket, esp)
+
+# Initialize a new MQTT Client object
+mqtt_client = MQTT.MQTT(
+    broker="io.adafruit.com", username=secrets["aio_user"], password=secrets["aio_key"],
+)
+
+
+# Initialize an Adafruit IO MQTT Client
+io = IO_MQTT(mqtt_client)
+
+# Connect the callback methods defined above to Adafruit IO
+io.on_connect = connected
+io.on_disconnect = disconnected
+
+# Connect to Adafruit IO
+print("Connecting to Adafruit IO...")
+io.connect()
+
+# Subscribe to the air quality sensor group
+group_air_quality = "air-quality-sensor"
+io.subscribe(group_key=group_air_quality)
+
+# Feeds within the air quality sensor group
+# Temperature
+feed_temp = group_air_quality + ".temperature"
+# Humidity
+feed_humid = group_air_quality + ".humidity"
+# Air quality index (AQI)
+feed_aqi = group_air_quality + ".aqi"
+# Air quality index category
+feed_aqi_category = group_air_quality + ".category"
 
 
 while True:
-    # TODO: Sample every 10min.
-    # Air Quality
+  try:
+    # Keep device connected to io.adafruit.com
+    # and process any incoming data.
+    io.loop()
+    # TODO: read every 10min
+    # air quality
     aqi_reading = sample_aq_sensor()
-    aqi_index, aqi_condition = calculate_aqi(aqi_reading)
-    print("AQI: %d"%aqi_index)
-    print("Condition: %s"%aqi_condition)
+    aqi, aqi_category = calculate_aqi(aqi_reading)
+    print("AQI: %d"%aqi)
+    print("category: %s"%aqi_category)
     # temp and humidity
     temperature, humidity = read_bme280()
     print("Temperature: %0.1f F" % temperature)
     print("Humidity: %0.1f %%" % humidity)
+  except (ValueError, RuntimeError) as e:
+      print("Failed to get data, retrying\n", e)
+      wifi.reset()
+      io.reconnect()
+      continue