sb-py/python-test/state_select.py

#!/usr/bin/env python

import uasyncio as asyncio
from workflows import workflows

class StateSelect:
    def __init__(self, lcd):
        self.lcd = lcd

    def enter(self, val = None):
        self.client = val
        self.current = 0

    def exit(self):
        return self.client, workflows[self.current]

    def draw_list(self):
        for i, wf in enumerate(workflows):
            s1 = "{}".format(wf["name"])
            s2 = "by: {}".format(wf["author"])

            off = i * 25 + 30
            if off >= self.lcd.height:
                break

            c = self.lcd.white
            if self.current == i:
                c = self.lcd.red

            self.lcd.hline(0, off, self.lcd.width, self.lcd.blue)
            self.lcd.text(s1, 0, off + 2, c)
            self.lcd.text(s2, 0, off + 12, c)

    async def draw(self):
        self.lcd.text("Please select your Workflow", 0, 10, self.lcd.red)

        keys = self.lcd.buttons()

        if keys.once("y"):
            print("user abort")
            return 5 # disconnect
        elif keys.once("up"):
            if self.current > 0:
                self.current -= 1
        elif keys.once("down"):
            if self.current < (len(workflows) - 1):
                self.current += 1
        elif keys.once("enter"):
            return 1 # connect

        self.draw_list()

        return -1 # stay in this state