import matplotlib.pyplot as plt
import matplotlib.font_manager as font_manager
from machine import Pin, I2C,Timer
import ssd1306
import network
import time
from machine import Pin
import dht
import ujson
from umqtt.simple import MQTTClient
import siot
import threading   #线程模块
import tkinter as tk
import json
import random

# ESP32 Pin assignment 
i2c = I2C(0, scl=Pin(22), sda=Pin(21))

oled_width = 128
oled_height = 64
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)

MQTT_CLIENT_ID = "wangchenyang"
MQTT_BROKER    = "broker.mqttdashboard.com"
MQTT_USER      = ""
MQTT_PASSWORD  = ""
MQTT_TOPIC     = "wangchenyang2306023125wsd"
TOPIC1 = b"ledctl2306023125"  #灯控主题，为了避免主题和其他同学一样，末尾需要添加自己的学号
TOPIC2 = b"pwmled2306023125"
TOPIC3 = b"ledstatus2306023125"
prev_weather = ""
d = dht.DHT22(Pin(15))
def f(t):
    d.measure()
    a=d.temperature()
    b=d.humidity()
    print('temperature:%2.1f dC'%a)
    print('humidity:%2.1f %%'%b)
    oled.fill(0)
    oled.text('wangchenyang', 10, 10)  
    oled.text('TEMP:%2.1f dC'%a,10,30)
    oled.text('HUMi:%2.1f %%'%b,10,50)
    oled.show()
tim=Timer(-1)
tim.init(period=2000,mode=Timer.PERIODIC,callback=f)

sensor = dht.DHT22(Pin(15))

print("Connecting to WiFi", end="")
sta_if = network.WLAN(network.STA_IF)# 选择gpio2引脚
led_blue = machine.PWM(machine.Pin(18), freq=1000)  # 设置 GPIO2 为输出
led_blue.duty(1023)

#连接wifi配置
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("Wifi Connected!")

# 当客户端连接到代理服务器时调用的回调函数
def on_connect(client, userdata, flags, rc):
    print("Connected with result code "+str(rc))
    client.subscribe("wcy+25+相关控制标识")

# 当接收到消息时调用的回调函数
def on_message(client, userdata, msg):
    print(msg.topic+" "+str(msg.payload))

client = mqtt.Client()
client.on_connect = on_connect
client.on_message = on_message

# 连接到MQTT代理服务器，这里需要替换为实际的代理服务器地址和端口
client.connect("broker.emqx.io", 1883, 60)

# 开始循环，保持连接并处理消息
client.loop_forever()

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:
  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")
    time.sleep(1)