status_led.py

import machine
import uasyncio as asyncio


class LedState:
    def __init__(self, r, g, b, blink):
        self.r = r
        self.g = g
        self.b = b
        self.blink = blink

    def __str__(self):
        return "{{ r: {}, g: {}, b: {}, blink: {}}}".format(
            self.r, self.g, self.b, self.blink
        )


class WmState:
    Undefined = LedState(1, 1, 1, True)

    ConnectingToNetwork = LedState(1, 1, 0, False)
    InitialisingSensors = LedState(0, 1, 1, True)
    ConnectingToMqtt = LedState(0, 0, 1, True)
    Idle = LedState(0, 0, 1, False)
    Publishing = LedState(0, 1, 0, False)

    LowBattery = LedState(1, 0, 0, True)
    Error = LedState(1, 0, 0, False)


class StatusLed:
    def __init__(self, led_pins):
        self.r_pin = machine.Pin(led_pins[0], machine.Pin.OUT)
        self.g_pin = machine.Pin(led_pins[1], machine.Pin.OUT)
        self.b_pin = machine.Pin(led_pins[2], machine.Pin.OUT)

        self.state = None
        self.warning_state = None
        self.set_state(WmState.Undefined)

    def __set_leds_on(self, state):
        self.r_pin.on() if state.r else self.r_pin.off()
        self.g_pin.on() if state.g else self.g_pin.off()
        self.b_pin.on() if state.b else self.b_pin.off()

    def __set_leds_off(self):
        self.r_pin.off()
        self.g_pin.off()
        self.b_pin.off()

    def set_state(self, state):
        # Turn LEDs on instantly regardless of `start_blink_loop` so that if
        # the calling code is blocking, the color still changes
        if state is not self.state:
            self.state = state
            self.__set_leds_on(state) if state else self.__set_leds_off()

    def set_warning_state(self, state):
        if state is not self.warning_state:
            self.warning_state = state
            self.__set_leds_on(state) if state else self.__set_leds_off()

    async def start_blink_loop(self):
        while True:
            # A state flagged as a warning takes priority
            if self.warning_state:
                self.__set_leds_on(self.warning_state)
                await asyncio.sleep_ms(200)

                # Need to check that self.warning_state is still not None
                # because it may have been changed during the sleep
                if self.warning_state and self.warning_state.blink:
                    self.__set_leds_off()
                    await asyncio.sleep_ms(200)

            else:
                self.__set_leds_on(self.state)
                await asyncio.sleep_ms(200)

                if self.state and self.state.blink:
                    self.__set_leds_off()
                    await asyncio.sleep_ms(200)