forked from ernitron/uPython-switch-control
/
httpserver.py
165 lines (146 loc) · 5.63 KB
/
httpserver.py
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
# Micropython Http Server
# Erni Tron ernitron@gmail.com
# Copyright (c) 2016
# Global import
import socket # Networking support
import time # Current time
import sys
import gc
import machine
# Local import
from request import parse_request
from config import config
from content import cb_open, cb_status, cb_getconf, cb_setconf, cb_resetconf
from content import cb_temperature, cb_temperature_json
from control import control
# A simple HTTP server
class Server:
def __init__(self, title='uServer'):
# Constructor
self.title = config.get_config('place')
self.conn = None
self.addr = None
self.head2 = cb_open('header.txt')
self.footer = cb_open('footer.txt')
self.timeout = 0.30
def activate(self, port, host='0.0.0.0'):
# Attempts to aquire the socket and launch the server
try:
self.socket = socket.socket()
self.socket.settimeout(self.timeout) # otherwise it will wait forever
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.socket.bind((host, port))
print("Server ", host, ":", port)
except Exception as e:
print(e)
self.socket.listen(1) # maximum number of queued connections
def wait_connections(self, interface, sleeptime):
# Main loop awaiting connections
refresh30 = '<meta http-equiv="refresh" content="300">\r\n'
redirect2 = '<meta http-equiv="refresh" content="2; url=/"/>\r\n'
error404 = '404 - Error'
while True:
try:
self.conn, self.addr = self.socket.accept()
except KeyboardInterrupt:
return
except:
continue
try:
req = self.conn.readline()
except:
continue
# conn - socket to client // addr - clients address
while True:
h = self.conn.readline()
if not h or h == b'\r\n':
break
# Some defaults that can be changed in progress
code = 200
extension = 'h'
refresh = ''
# determine request method (GET / POST are supported)
r = parse_request(req)
if r == None:
code = 404
content = error404
elif r['uri'] == b'/' or r['uri'] == b'/index' :
content = '<h2>%s %s</h2>' % (self.title, control.status())
elif r['uri'] == b'/on' :
content = '<h2>Switch %s</h2>' % control.switch_on()
elif r['uri'] == b'/off' :
content = '<h2>Switch %s</h2>' % control.switch_off()
elif r['uri'] == b'/temperature' :
content = cb_temperature()
content = ''
elif r['uri'] == b'/j' :
extension = 'j'
content = cb_temperature_json()
content = ''
elif r['uri'] == b'/getconf':
content = cb_getconf()
elif b'/setconf' in r['uri']:
if 'key' in r['args'] and 'value' in r['args']:
content = cb_setconf(r['args']['key'], r['args']['value'])
elif 'key' in r['args'] :
content = cb_setconf(r['args']['key'], None)
else:
content = cb_setconf(None, None)
self.title = config.get_config('place') # just in case
elif r['uri'] == b'/status':
content = cb_status()
elif r['uri'] == b'/webrepl' :
import webrepl
webrepl.start()
sleeptime = 299 # almost 5 minutes
refresh = redirect2
code = 302
content = '<h2>Webrepl +5min</h2></div>'
elif r['uri'] == b'/5min' :
sleeptime = 299 # almost 5 minutes
refresh = redirect2
code = 302
content = '<h2>+5min</h2></div>'
elif r['uri'] == b'/reboot' :
content = '<h2>Reboot</h2></div>'
self.http_send(code, content, extension, redirect2)
self.conn.close()
machine.reset()
elif r['file'] != b'':
myfile = r['file']
code = 200
content = cb_open(myfile)
else:
code = 404
content = error404
# At end of loop just close socket and collect garbage
self.http_send(code, content, extension, refresh)
self.conn.close()
gc.collect()
def http_send(self, code, content, extension='h', refresh=''):
mt = {'h': "text/html", 'j': "application/json"}
codes = {200:" OK", 400:" Bad Request", 404:" Not Found", 302:" Redirect", 501:"Server Error" }
head0 = 'HTTP/1.1 %s\r\nServer: tempserver\r\nContent-Type: %s\r\n'
#head1 = 'Cache-Control: private, no-store\r\nConnection: close\r\n\r\n'
head1 = 'Connection: close\r\n\r\n'
if code not in codes: code = 501
httpstatus = str(code) + codes[code]
if extension not in mt: extension = 'h'
mimetype = mt[extension]
head0 = head0 % (httpstatus, mimetype)
self.conn.send(head0)
if refresh:
self.conn.send(refresh)
self.conn.send(head1)
if extension != 'j':
for c in self.head2:
self.conn.send(c)
# if type(content) is list or type(content) is tuple:
if type(content) is list :
for c in content:
self.conn.send(c)
elif content != '':
self.conn.send(content)
if extension != 'j':
for c in self.footer:
self.conn.sendall(c)