mirror
/
micropython-WiFiManager
kopia lustrzana https://github.com/tayfunulu/WiFiManager
1
0
Forkuj 0
Kod Zgłoszenia Projekty Wydania Wiki Aktywność

Deleted extra empty lines 

Deleted extra empty lines
pull/14/head
Tayfun ULU 2018-05-25 23:23:52 +03:00 zatwierdzone przez GitHub
rodzic a1f955b338
commit 9d3094862a
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 4 dodań i 29 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

33
wifimgr.py
Wyświetl plik

@ -3,7 +3,6 @@ import socket
import ure import ure
import time import time
ap_ssid = "WifiManager" ap_ssid = "WifiManager"
ap_password = "tayfunulu" ap_password = "tayfunulu"
ap_authmode = 3 # WPA2 ap_authmode = 3 # WPA2
@ -105,25 +104,21 @@ def send_header(client, status_code=200, content_length=None ):
client.sendall("Content-Type: text/html\r\n") client.sendall("Content-Type: text/html\r\n")
if content_length is not None: if content_length is not None:
client.sendall("Content-Length: {}\r\n".format(content_length)) client.sendall("Content-Length: {}\r\n".format(content_length))
client.sendall("\r\n") client.sendall("\r\n")
def send_response(client, payload, status_code=200): def send_response(client, payload, status_code=200):
content_length = len(payload) content_length = len(payload)
send_header(client, status_code, content_length) send_header(client, status_code, content_length)
if content_length > 0: if content_length > 0:
client.sendall(payload) client.sendall(payload)
client.close() client.close()
def handle_root(client): def handle_root(client):
wlan_sta.active(True) wlan_sta.active(True)
ssids = sorted(ssid.decode('utf-8') for ssid, *_ in wlan_sta.scan()) ssids = sorted(ssid.decode('utf-8') for ssid, *_ in wlan_sta.scan())
send_header(client) send_header(client)
client.sendall("""\ client.sendall("""\
<html> <html>
<h1 style="color: #5e9ca0; text-align: center;"> <h1 style="color: #5e9ca0; text-align: center;">
@ -135,10 +130,8 @@ def handle_root(client):
<table style="margin-left: auto; margin-right: auto;"> <table style="margin-left: auto; margin-right: auto;">
<tbody> <tbody>
""") """)
while len(ssids): while len(ssids):
ssid = ssids.pop(0) ssid = ssids.pop(0)
client.sendall("""\ client.sendall("""\
<tr> <tr>
<td colspan="2"> <td colspan="2">
@ -146,7 +139,6 @@ def handle_root(client):
</td> </td>
</tr> </tr>
""".format(ssid)) """.format(ssid))
client.sendall("""\ client.sendall("""\
<tr> <tr>
<td>Password:</td> <td>Password:</td>
@ -183,12 +175,11 @@ def handle_root(client):
</ul> </ul>
</html> </html>
""" % dict(filename=NETWORK_PROFILES)) """ % dict(filename=NETWORK_PROFILES))
client.close() client.close()
def handle_configure(client, request): def handle_configure(client, request):
match = ure.search("ssid=([^&]*)&password=(.*)", request) match = ure.search("ssid=([^&]*)&password=(.*)", request)
if match is None: if match is None:
send_response(client, "Parameters not found", status_code=400) send_response(client, "Parameters not found", status_code=400)
return False return False
@ -199,11 +190,9 @@ def handle_configure(client, request):
except: except:
ssid = match.group(1).replace("%3F", "?").replace("%21", "!") ssid = match.group(1).replace("%3F", "?").replace("%21", "!")
password = match.group(2).replace("%3F", "?").replace("%21", "!") password = match.group(2).replace("%3F", "?").replace("%21", "!")
if len(ssid) == 0: if len(ssid) == 0:
send_response(client, "SSID must be provided", status_code=400) send_response(client, "SSID must be provided", status_code=400)
return False return False
if do_connect(ssid, password): if do_connect(ssid, password):
response = """\ response = """\
<html> <html>
@ -252,7 +241,6 @@ def handle_not_found(client, url):
def stop(): def stop():
global server_socket global server_socket
if server_socket: if server_socket:
server_socket.close() server_socket.close()
server_socket = None server_socket = None
@ -260,59 +248,46 @@ def stop():
def start(port=80): def start(port=80):
global server_socket global server_socket
addr = socket.getaddrinfo('0.0.0.0', port)[0][-1] addr = socket.getaddrinfo('0.0.0.0', port)[0][-1]
stop() stop()
wlan_sta.active(True) wlan_sta.active(True)
wlan_ap.active(True) wlan_ap.active(True)
wlan_ap.config(essid=ap_ssid, password=ap_password, authmode=ap_authmode) wlan_ap.config(essid=ap_ssid, password=ap_password, authmode=ap_authmode)
server_socket = socket.socket() server_socket = socket.socket()
server_socket.bind(addr) server_socket.bind(addr)
server_socket.listen(1) server_socket.listen(1)
print('Connect to WiFi ssid ' + ap_ssid + ', default password: ' + ap_password) print('Connect to WiFi ssid ' + ap_ssid + ', default password: ' + ap_password)
print('and access the ESP via your favorite web browser at 192.168.4.1.') print('and access the ESP via your favorite web browser at 192.168.4.1.')
print('Listening on:', addr) print('Listening on:', addr)
while True: while True:
if wlan_sta.isconnected(): if wlan_sta.isconnected():
return True return True
client, addr = server_socket.accept() client, addr = server_socket.accept()
print('client connected from', addr) print('client connected from', addr)
try: try:
client.settimeout(5.0) client.settimeout(5.0)
request = b"" request = b""
try: try:
while "\r\n\r\n" not in request: while "\r\n\r\n" not in request:
request += client.recv(512) request += client.recv(512)
except OSError: except OSError:
pass pass
print("Request is: {}".format(request)) print("Request is: {}".format(request))
if "HTTP" not in request: if "HTTP" not in request:
# skip invalid requests # skip invalid requests
continue continue
# version 1.9 compatibility # version 1.9 compatibility
try: try:
url = ure.search("(?:GET|POST) /(.*?)(?:\\?.*?)? HTTP", request).group(1).decode("utf-8").rstrip("/") url = ure.search("(?:GET|POST) /(.*?)(?:\\?.*?)? HTTP", request).group(1).decode("utf-8").rstrip("/")
except: except:
url = ure.search("(?:GET|POST) /(.*?)(?:\\?.*?)? HTTP", request).group(1).rstrip("/") url = ure.search("(?:GET|POST) /(.*?)(?:\\?.*?)? HTTP", request).group(1).rstrip("/")
print("URL is {}".format(url)) print("URL is {}".format(url))
if url == "": if url == "":
handle_root(client) handle_root(client)
elif url == "configure": elif url == "configure":
handle_configure(client, request) handle_configure(client, request)
else: else:
handle_not_found(client, url) handle_not_found(client, url)
finally: finally:
client.close() client.close()