nodemcu/flexible_web_server/main.py at d311d61a660da1dbedc5893ebd6dcf1f3c745312 · pcewebpython/nodemcu · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
try:
    import usocket as socket
except:
    import socket

response_404 = """HTTP/1.0 404 NOT FOUND
<h1>404 Not Found</h1>
"""

response_500 = """HTTP/1.0 500 INTERNAL SERVER ERROR
<h1>500 Internal Server Error</h1>
"""
response_template = """HTTP/1.0 200 OK

%s
"""
import machine
import ntptime, utime
from machine import RTC
from time import sleep

seconds = ntptime.time()
rtc = RTC()
rtc.datetime(utime.localtime(seconds))

adc = machine.ADC(0)
pin = machine.Pin(16, machine.Pin.OUT)
switch_pin = machine.Pin(10, machine.Pin.IN)


def time():
    body = """<html>
 <body>
 <h1>Time</h1>
 <p>%s</p>
 </body>
 </html>
    """ % str(rtc.datetime())

    return response_template % body


def dummy():
    body = "This is a dummy endpoint"

    return response_template % body


def light():
    body = "{value: " + adc.read() + "}"
    return response_template % body


def switch():
    body = "{state: " + switch_pin.value() + "}"
    return response_template % body


def light_on():
     pin.value(0)
     body = "You turned a light on!"
     return response_template % body


def light_off():
     pin.value(1)
     body = "You turned a light off!"
     return response_template % body


handlers = {
    'time': time,
    'dummy': dummy,
    'light_on': light_on,
    'light_off': light_off,
    'light': light,
    'switch': switch,
}


def main():
    s = socket.socket()
    ai = socket.getaddrinfo("0.0.0.0", 8080)
    addr = ai[0][-1]

    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

    s.bind(addr)
    s.listen(5)
    print("Listening, connect your browser to http://<this_host>:8080/")

    while True:
        sleep(1)
        res = s.accept()
        client_s = res[0]
        client_addr = res[1]
        req = client_s.recv(4096)
        print("Request:")
        print(req)
        try:
            # The first line of a request looks like "GET /arbitrary/path/ HTTP/1.1".
            # This grabs that first line and whittles it down to just "/arbitrary/path/"
            path = req.decode().split("\r\n")[0].split(" ")[1]

            # Given the path, identify the correct handler to use
            handler = handlers[path.strip('/').split('/')[0]]
            response = handler()
        except KeyError:
            response = response_404
        except Exception as e:
            response = response_500
            print(str(e))

        # A handler returns an entire response in the form of a multi-line string.
        # This breaks up the response into single strings, byte-encodes them, and
        # joins them back together with b"\r\n". Then it sends that to the client.
        client_s.send(b"\r\n".join([line.encode() for line in response.split("\n")]))

        client_s.close()
        print()

main()