"""
MicroPython IoT Weather Station Example for Wokwi.com

To view the data:

1. Go to http://www.hivemq.com/demos/websocket-client/
2. Click "Connect"
3. Under Subscriptions, click "Add New Topic Subscription"
4. In the Topic field, type "wokwi-weather" then click "Subscribe"

Now click on the DHT22 sensor in the simulation,
change the temperature/humidity, and you should see
the message appear on the MQTT Broker, in the "Messages" pane.

Copyright (C) 2022, Uri Shaked

https://wokwi.com/arduino/projects/322577683855704658
"""

import network
import time
from machine import Pin
import dht
import ujson
from umqtt.simple import MQTTClient
import random

# MQTT Server Parameters
MQTT_CLIENT_ID = "micropython-weather-demo"
MQTT_BROKER    = "broker.mqttdashboard.com"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "wokwi-weather"

sensor = dht.DHT22(Pin(15))
p0 = Pin(2, Pin.OUT)
p1 = Pin(4, Pin.OUT)
p2 = Pin(5, Pin.OUT)

print("Connecting to 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(" Connected!")

print("Connecting to MQTT server... ", end="")
client = MQTTClient(MQTT_CLIENT_ID, MQTT_BROKER, user=MQTT_USER, password=MQTT_PASSWORD)
client.connect()

print("Connected!")

prev_weather = ""
while True:
  p0.value(0)
  p1.value(0)
  p2.value(0)
  print("Measuring weather conditions... ", end="")
  sensor.measure() 
  message = ujson.dumps({
    "temp": sensor.temperature(),
    "humidity": sensor.humidity(),
  })

  if message != prev_weather:
    print("Updated!")
    print("Reporting to MQTT topic {}: {}".format(MQTT_TOPIC, message))
    client.publish(MQTT_TOPIC, message)
    prev_weather = message
  else:
    print("No change")

  h = float(message[12:16])
  if (0 <= h <10):
    p0.value(0)
    p1.value(0)
    p2.value(0)
  elif (10 <= h <20):
    p0.value(1)
    p1.value(0)
    p2.value(0)
  elif (20 <= h <30):
    p0.value(0)
    p1.value(1)
    p2.value(0)
  elif (30 <= h <40):
    p0.value(1)
    p1.value(1)
    p2.value(0)
  elif (40 <= h <50):
    p0.value(0)
    p1.value(0)
    p2.value(1)
  elif (50 <= h <60):
    p0.value(1)
    p1.value(0)
    p2.value(1)
  elif (60 <= h <70):
    p0.value(0)
    p1.value(1)
    p2.value(1)
  else:
    p0.value(1)
    p1.value(1)
    p2.value(1)


  time.sleep(5)