import network
import urequests
import time
import json

# ====== ตั้งค่าการเชื่อมต่อ WiFi ======
SSID = "Wokwi-GUEST"
PASSWORD = ""

def connect_wifi():
    wlan = network.WLAN(network.STA_IF)
    wlan.active(True)
    wlan.connect(SSID, PASSWORD)
    
    print("กำลังเชื่อมต่อ WiFi...")
    while not wlan.isconnected():
        time.sleep(1)
    print("เชื่อมต่อสำเร็จ!")
    print("IP:", wlan.ifconfig()[0])

# ====== ดึงข้อมูล AQI ======
def get_aqi_data():
    url = "http://air4thai.pcd.go.th/services/getNewAQI_JSON.php?region=2"
    try:
        resp = urequests.get(url)
        data = resp.json()
        resp.close()
        return data
    except Exception as e:
        print("Error:", e)
        return None

# ====== หา สถานี ลำปาง ======
def find_lampang_station(aqi_data):
    if not aqi_data:
        return None
    stations = aqi_data.get("stations", [])
    print(stations)

    #for s in stations:
    #    name = s.get("nameTH", "")
    #    if "สถานีอุตุนิยมวิทยาลำปาง" in name:
            #print(s)
    #        return s

    for s in stations:
        name = s.get("stationID", "")
        print(name)
        if "40t" in name:
            #print(s)
            return s


    return None


# ====== แสดงผล ======
def show_pm25(st):
    if not st:
        print("ไม่พบสถานีลำปาง")
        return

    aqi = st.get("AQILast", {})
    pm25 = aqi.get("PM25", {})

    print("-----------------------")
    print("สถานี :", st.get("nameTH"))
    print("พื้นที่ :", st.get("areaTH"))
    print("วันที่ :", aqi.get("date"), aqi.get("time"))
    print("PM2.5 :", pm25.get("value"), "µg/m³")
    print("AQI   :", pm25.get("aqi"))
    print("-----------------------")

# ====== Main ======
def main():
    connect_wifi()
    
    print("\nกำลังดึงข้อมูล AQI...")
    aqi_data = get_aqi_data()

    lampang = find_lampang_station(aqi_data)
    show_pm25(lampang)

if __name__ == "__main__":
    main()