Fixed clock drift and added setup wizard
This commit is contained in:
162
firmware/main.py
Normal file
162
firmware/main.py
Normal file
@@ -0,0 +1,162 @@
|
||||
import gc
|
||||
import network
|
||||
import picoweb
|
||||
import json
|
||||
from time import sleep_ms
|
||||
from timezone import TIMEZONES
|
||||
|
||||
app = picoweb.WebApp(__name__)
|
||||
ap_if = network.WLAN(network.AP_IF)
|
||||
sta_if = network.WLAN(network.STA_IF)
|
||||
nets = sta_if.scan()
|
||||
sta_if.active(True)
|
||||
config = dict()
|
||||
try:
|
||||
with open("config.json") as fh:
|
||||
config = json.loads(fh.read())
|
||||
sta_if.connect(config.get("ssid"), config.get("password"))
|
||||
except OSError:
|
||||
pass
|
||||
|
||||
print("Scanning for wireless networks...")
|
||||
|
||||
@app.route("/connect")
|
||||
def index(req, resp):
|
||||
if req.method == "POST":
|
||||
yield from req.read_form_data()
|
||||
else:
|
||||
req.parse_qs()
|
||||
yield from picoweb.start_response(resp)
|
||||
with open("config.json", "w") as fh:
|
||||
fh.write(json.dumps(req.form))
|
||||
yield from resp.awrite("Setting saved please power cycle device")
|
||||
|
||||
|
||||
@app.route("/")
|
||||
def index(req, resp):
|
||||
print("Serving index")
|
||||
yield from picoweb.start_response(resp)
|
||||
yield from resp.awrite("<html>")
|
||||
yield from resp.awrite("<head>")
|
||||
yield from resp.awrite("<meta name='viewport' content='width=device-width, initial-scale=1'>")
|
||||
yield from resp.awrite("</head>")
|
||||
yield from resp.awrite("<body>")
|
||||
yield from resp.awrite("<h>Welcome to NixiESP12 configuration wizard</h>")
|
||||
yield from resp.awrite("<p>Detected wireless networks:</p>")
|
||||
yield from resp.awrite("<form action='/connect' method='post'>")
|
||||
yield from resp.awrite("<p>Select wireless network:</p>")
|
||||
yield from resp.awrite("<select name='ssid'>")
|
||||
for ssid, _, _, snr, crypto, _ in nets:
|
||||
ssid = ssid.decode("utf-8")
|
||||
yield from resp.awrite("<option>" + ssid + "</option>")
|
||||
yield from resp.awrite("</select>")
|
||||
yield from resp.awrite("<p>Wireless password is applicable:</p>")
|
||||
yield from resp.awrite("<input type='password' name='password'/>")
|
||||
yield from resp.awrite("<p>Timezone:</p>")
|
||||
yield from resp.awrite("<select name='timezone'>")
|
||||
for index, (dst, offset, title) in enumerate(TIMEZONES):
|
||||
yield from resp.awrite("<option value='%d'>%s</option>" % (index, title))
|
||||
yield from resp.awrite("</select>")
|
||||
yield from resp.awrite("<p>NTP resynchronization interval:</p>")
|
||||
yield from resp.awrite("<select name='interval'>")
|
||||
for j in range(0, 73):
|
||||
yield from resp.awrite("<option value='%d'>%d hours</option>" % (j*3600, j))
|
||||
yield from resp.awrite("</select>")
|
||||
yield from resp.awrite("<p> </p>")
|
||||
yield from resp.awrite("<input type='submit'/>")
|
||||
yield from resp.awrite("</form>")
|
||||
yield from resp.awrite("</body>")
|
||||
yield from resp.awrite("<html>")
|
||||
|
||||
timed_out = True
|
||||
if config:
|
||||
print("Connecting to", config.get("ssid"))
|
||||
for j in range(0,30):
|
||||
if sta_if.isconnected():
|
||||
ap_if.active(False)
|
||||
timed_out = False
|
||||
break
|
||||
sleep_ms(200)
|
||||
|
||||
if timed_out:
|
||||
ap_if.active(True)
|
||||
print("Starting setup wizard on", ap_if.ifconfig())
|
||||
app.run()
|
||||
|
||||
|
||||
TIMEZONE = TIMEZONES[int(config.get("timezone", 30))][1]
|
||||
print("Using timezone", TIMEZONES[int(config.get("timezone", 30))])
|
||||
RESYNC = int(config.get("interval")) # Resync once in 8 hours
|
||||
print("NTP resynchronization interval", RESYNC, "seconds")
|
||||
DEBUG = False
|
||||
|
||||
print("Press Ctrl-C now abort main.py execution and to retain keyboard input")
|
||||
sleep_ms(2000)
|
||||
|
||||
import time
|
||||
import ntptime
|
||||
from machine import Pin, Timer
|
||||
|
||||
# Note that keyboard input is lost beyond this point!
|
||||
clock = Pin(3, mode=Pin.OUT)
|
||||
latch = Pin(0, mode=Pin.OUT)
|
||||
data = Pin(2, mode=Pin.OUT)
|
||||
blink = False
|
||||
lookup = 11, 9, 12, 8, 0, 4, 1, 3, 2, 10
|
||||
countdown = 0
|
||||
|
||||
def bitbang_bit(value):
|
||||
if value & 1:
|
||||
data.on()
|
||||
else:
|
||||
data.off()
|
||||
clock.on()
|
||||
clock.off()
|
||||
|
||||
def bitbang_digit(digit):
|
||||
bitbang_bit(blink)
|
||||
for i in range(0,4):
|
||||
bitbang_bit(lookup[digit] << i >> 3)
|
||||
bitbang_bit(blink)
|
||||
bitbang_bit(blink)
|
||||
bitbang_bit(blink)
|
||||
|
||||
def dump_time(hour, minute, second):
|
||||
if DEBUG:
|
||||
print("Time is %02d:%02d:%02d" % (hour, minute, second))
|
||||
bitbang_digit(hour // 10)
|
||||
bitbang_digit(hour % 10)
|
||||
bitbang_digit(minute // 10)
|
||||
bitbang_digit(minute % 10)
|
||||
bitbang_digit(second // 10)
|
||||
bitbang_digit(second % 10)
|
||||
|
||||
# RTC accuracy is still garbage, time.ticks_ms() which is bound to CPU ticks seems to be more accurate
|
||||
# https://forum.micropython.org/viewtopic.php?t=3251#p19092
|
||||
|
||||
while True:
|
||||
if countdown <= 0:
|
||||
try:
|
||||
ticks_then, time_then = time.ticks_ms(), ntptime.time()
|
||||
except OSError:
|
||||
print("Resync failed")
|
||||
else:
|
||||
countdown = RESYNC
|
||||
print("Resync done")
|
||||
else:
|
||||
year, month, day, hour, minute, second, _, _ = time.localtime(time_then + (time.ticks_ms() - ticks_then) // 1000)
|
||||
sleep_ms(500-(time.ticks_ms() - ticks_then) % 1000)
|
||||
blink = True
|
||||
dump_time((hour + TIMEZONE) % 24, minute, second)
|
||||
latch.on()
|
||||
latch.off()
|
||||
countdown -= 1
|
||||
|
||||
year, month, day, hour, minute, second, _, _ = time.localtime(time_then + (time.ticks_ms() - ticks_then) // 1000)
|
||||
sleep_ms(1001-(time.ticks_ms() - ticks_then) % 1000)
|
||||
blink = False
|
||||
dump_time((hour + TIMEZONE) % 24, minute, second)
|
||||
latch.on()
|
||||
latch.off()
|
||||
|
||||
main()
|
||||
Reference in New Issue
Block a user