nixiesp12/firmware/main.py

163 lines
5.1 KiB
Python

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>&nbsp;</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()