import gc import network import time import socket from st3m.application import Application, ApplicationContext from umqtt.robust import MQTTClient WIFI_SSID = b'...' WIFI_PSK = b'...' MQTT_BROKER_IP = '...' MQTT_CLIENT_ID = b'flow3r-{}' MQTT_BASE_TOPIC = b'flow3r-{}/event' 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 = MQTT_CLIENT_ID.format(self.nic.config('mac').hex()) self.base_topic = MQTT_BASE_TOPIC.format(self.nic.config('mac').hex()) self.mqtt = MQTTClient(client_id, 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(WIFI_SSID, 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()))