import network
import socket
import machine

ssid = '荣耀-Andy'
password = 'x6446547920.'

def connect_wifi():
    wlan = network.WLAN(network.STA_IF)
    wlan.active(True)
    wlan.connect(ssid, password)
    while not wlan.isconnected():
        pass
    return wlan.ifconfig()[0]

def create_server(ip):
    s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    s.bind(('', 80))
    s.listen(5)
    return s

def handle_client(client):
    request = client.recv(1024)
    if b'/on' in request:
        machine.Pin(2, machine.Pin.OUT).value(1)
    elif b'/off' in request:
        machine.Pin(2, machine.Pin.OUT).value(0)
    client.send('HTTP/1.1 200 OK\n\n')
    client.close()

def main():
    ip = connect_wifi()
    server = create_server(ip)
    print('Server started at', ip)
    while True:
        client, addr = server.accept()
        handle_client(client)

main()