flow3r-openhab/flow3r_openhab/__init__.py

import gc
import json
import network
import os
import socket
import time

from st3m.application import Application, ApplicationContext

from umqtt.robust import MQTTClient


with open(os.path.join(os.path.dirname(__file__), 'config.json'), 'r') as f:
    CONFIG = json.load(f)


class Flow3rOpenhabMqtt(Application):

    def __init__(self, app_ctx: ApplicationContext) -> None:
        super().__init__(app_ctx)
        self.state = 0
        self.nic = network.WLAN(network.STA_IF)
        client_id = CONFIG['mqtt_client_id'].format(self.nic.config('mac').hex())
        self.base_topic = CONFIG['mqtt_topic'].format(self.nic.config('mac').hex())
        self.mqtt = MQTTClient(client_id, CONFIG['mqtt_broker_ip'])
        self.touch_time = [None]*10
        self.touch_begin = [None]*10
        self.touch_last = [None]*10
        self.touch_base = [False]*10
        self.touch_tip = [False]*10

    def draw(self, ctx: Context) -> None:
        if self.state == 0:
            return
        ctx.line_width = 18
        ctx.line_join = ctx.BEVEL
        ctx.rgb(0, 0, 0).rectangle(-120, -120, 240, 240).fill()
        ctx.rgb(0.3, 0.3, 0.3).arc(0, 0, 90, 0, 2.39, 0).arc(0, 0, 90, 2.79, 6.29, 0).stroke()
        ctx.rgb(0.9, 0.3, 0.1).move_to(-96, 53).line_to(-1, -43).line_to(71, 29).stroke()

    def do_connect(self):
        self.nic.active(True)
        if not self.nic.isconnected():
            print('connecting to network...')
            self.nic.connect(CONFIG['wifi_ssid'], CONFIG['wifi_psk'])
            while not self.nic.isconnected():
                pass
        print('network config:', self.nic.ifconfig())

    def think(self, ins: InputState, delta_ms: int) -> None:
        if self.nic.status() not in [network.STAT_CONNECTING, network.STAT_GOT_IP]:
            self.do_connect()
        if self.nic.status() == network.STAT_GOT_IP and self.state == 0:
            gc.collect()
            self.mqtt.connect()
            self.state = 1
        if self.state == 1:
            for i in range(10):
                if ins.captouch.petals[i].pressed:
                    if self.touch_begin[i] is None:
                        self.touch_begin[i] = ins.captouch.petals[i].position
                        self.touch_time[i] = time.ticks_ms()
                    self.touch_last[i] = ins.captouch.petals[i].position
                    self.touch_base[i] = ins.captouch.petals[i].pads.base
                    if hasattr(ins.captouch.petals[i].pads, 'tip'):
                        self.touch_tip[i] = ins.captouch.petals[i].pads.tip
                    else:
                        self.touch_tip[i] = ins.captouch.petals[i].pads.cw or ins.captouch.petals[i].pads.ccw
                else:
                    if self.touch_begin[i] is not None:
                        duration = time.ticks_ms() - self.touch_time[i]
                        d_x = self.touch_last[i][0] - self.touch_begin[i][0]
                        d_y = self.touch_last[i][1] - self.touch_begin[i][1]
                        event = None
                        if abs(d_x) < 25000 and abs(d_y) < 25000:
                            if self.touch_base[i]:
                                event = 'touch_base'
                            elif self.touch_tip[i]:
                                event = 'touch_tip'
                        elif abs(d_x) > abs(d_y):
                            if d_x > 0:
                                event = 'swipe_up'
                            else:
                                event = 'swipe_down'
                        else:
                            if d_y > 0:
                                event = 'swipe_right'
                            else:
                                event = 'swipe_left'
                        if event is not None:
                            human_petal = 10-(-i%10)
                            self.mqtt.publish(self.base_topic, b'{{"petal":"{}","event":"{}","dx":{},"dy":{},"duration":{}}}'.format(human_petal, event, d_x, d_y, duration))
                        self.touch_time[i] = None
                        self.touch_begin[i] = None
                        self.touch_last[i] = None
                        self.touch_base[i] = False
                        self.touch_tip[i] = False


if __name__ == "__main__":
    import st3m.run
    st3m.run.run_view(Flow3rOpenhabMqtt(ApplicationContext()))