# 连接WiFi
import network

wlan = network.WLAN(network.STA_IF) # 选择STA模式
wlan.active(True)       # 开始工作
wlan.scan()             # 扫描WiFi
wlan.isconnected()      # 判断是否连接
wlan.connect('essid', 'password') # 这个就是账号，密码
wlan.config('mac')      # 获取 MAC 地址
wlan.ifconfig()         # 获得信息

from socket import *

# 1. 创建udp套接字
udp_socket = socket(AF_INET, SOCK_DGRAM)

# 2. 准备接收方的地址
dest_addr = ('192.168.43.27', 8080)

# 3. 从键盘获取数据
send_data = "hello world"

# 4. 发送数据到指定的电脑上
udp_socket.sendto(send_data.encode('utf-8'), dest_addr)

# 开灯的代码
import network
import time
import socket
import machine


def do_connect():
    # 1. 连接wifi
    
    # 1.1. 设置为STA模式
    wlan = network.WLAN(network.STA_IF)
    # 1.2. 使其工作
    wlan.active(True)
    # 1.3. 是否连接工作
    if not wlan.isconnected():
        print('connecting to network...')
        # 1.4. 连接WiFi
        wlan.connect('esp', '123456798')
        i = 1
        while not wlan.isconnected():
            print("正在连接...{}".format(i))
            i += 1
            time.sleep(1)
    # 1.4. 输出相应信息
    print('network config:', wlan.ifconfig())
    
def start_udp():
    # 2. 启动网络功能（UDP）

    # 2.1. 创建udp套接字
    udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 2.2. 绑定本地信息
    udp_socket.bind(("0.0.0.0", 7788))

    return udp_socket
    
    
def main():
    # 1. 链接wifi
    do_connect()
    # 2. 创建UDP
    udp_socket = start_udp()
    # 3. 创建灯对象
    led = machine.Pin(2, machine.Pin.OUT)
    # 4. 接收网络数据
    while True:
        recv_data, sender_info = udp_socket.recvfrom(1024)
        # 4.1. 解码
        recv_data_str = recv_data.decode("utf-8")
        try:
            print(recv_data_str)
        except Exception as ret:
            print("error:", ret)
        
        # 5. 处理接收的数据
        if recv_data_str == "开灯":
            print("这是要开灯的代码...")
            led.value(1)
        elif recv_data_str == "关灯":
            print("这是要关灯的代码...")
            led.value(0)    
# 只有直接运行代码时才执行    
if __name__ == "__main__":
    main()