from machine import I2C, Pin
from i2c_lcd import *
import time
import dht
import network
import ujson
from umqtt.simple import MQTTClient

MQTT_CLIENT_ID = ""
MQTT_BROKER    = "broker.mqttdashboard.com"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "artemov"


def sub_cb(topic, msg):
    global x, y
    topic = topic.decode('utf-8')
    message = msg.decode('utf-8')
    print(topic, msg)
    if topic == MQTT_TOPIC and '/z ' in message:
        x, y = numbers = map(int, message.split()[1:])
        x, y = abs(x), abs(y)
        x -= 1
        y -= 1
        x %= 20
        y %= 4

sensor = dht.DHT22(Pin(22))

humpin = Pin(15, Pin.OUT)
tempin = Pin(17, Pin.OUT)
dht_interval = 500
hum=40
temp=40

i2c1 = SoftI2C(sda=Pin(12), scl=Pin(13))
lcd1 = LiquidCrystalI2C(i2c1, 0x27, 16, 2)
lcd1.set_controlled_cursor(False)
lcd1.home()
lcd1.display()
lcd1.backlight()

i2c2 = SoftI2C(sda=Pin(18), scl=Pin(19))
lcd2 = LiquidCrystalI2C(i2c2, 0x27, 20, 4)
lcd2.set_controlled_cursor(False)
lcd2.home()
lcd2.display()
lcd2.backlight()

print("Подключение к WiFi", end="")
sta_if = network.WLAN(network.STA_IF)
sta_if.active(True)
sta_if.connect('Wokwi-GUEST', '')
while not sta_if.isconnected():
  print(".", end="")
  time.sleep(0.1)
print(" Подключено!")

print("Поключение к серверу MQTT... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.set_callback(sub_cb)
client.connect()
client.subscribe(MQTT_TOPIC)

prev_weather = ""
x, y = 0, 0

while True:
    client.check_msg()

    sensor.measure()
    hum = sensor.humidity()
    temp = sensor.temperature()
    
    lcd1.clear()
    lcd1.print(f'Temp: {temp}')
    lcd1.set_cursor(0, 1)
    lcd1.print(f'Hum: {hum}')

    lcd2.clear()
    for i in range(20):
        # print(i, y)
        lcd2.set_cursor(i, y)
        lcd2.print("*")
    for j in range(4):
        # print(x, j)
        lcd2.set_cursor(x, j)
        lcd2.print("*")
  
    if hum < 20 or hum > 60:
        humpin.value(1)
    else:
        humpin.value(0)
    
    if temp > 45 or temp < 18:
        tempin.value(1)
    else:
        tempin.value(0)

    message = ujson.dumps({
        "temp": temp,
        "humidity": hum,
    })

    if message != prev_weather:
        client.publish(MQTT_TOPIC, message)
        prev_weather = message
    print(message)
    
    time.sleep(1)