85 lines
2.4 KiB
Python
85 lines
2.4 KiB
Python
import network
|
|
import ujson
|
|
import ure as re
|
|
import usocket as socket
|
|
import time
|
|
from gs.little_apache import LittleApache
|
|
|
|
|
|
class Setup:
|
|
|
|
wlans = []
|
|
|
|
def __init__(self):
|
|
self.ap_ssid = "Growsystem 1.0"
|
|
self.ap_password = "password"
|
|
self.wlans = []
|
|
self.selected_ssid = ""
|
|
self.wlan_password = ""
|
|
self.pin = ""
|
|
|
|
def scan_wlans(self):
|
|
print("Scan for WiFis")
|
|
wlan = network.WLAN(network.STA_IF)
|
|
wlan.active(True)
|
|
self.wlans = [w[0].decode() for w in wlan.scan()]
|
|
print("Detected WiFis: ", self.wlans)
|
|
wlan.active(False)
|
|
|
|
def start_ap_mode(self):
|
|
wlan = network.WLAN(network.STA_IF)
|
|
wlan.active(False)
|
|
wlan_config = {
|
|
"ssid": self.ap_ssid,
|
|
"pw": self.ap_password
|
|
}
|
|
print("Switch to ap mode with data:", wlan_config)
|
|
ap = network.WLAN(network.AP_IF)
|
|
ap.config(essid=self.ap_ssid, password=self.ap_password)
|
|
ap.active(True)
|
|
self.net = ap
|
|
print("Connect with your browser to:", ap.ifconfig()[0])
|
|
|
|
def stop_ap_mode(self):
|
|
print("Stop ap mode")
|
|
ap = network.WLAN(network.AP_IF)
|
|
ap.active(False)
|
|
|
|
def get_initial_config_webserver(self):
|
|
self.la = LittleApache(self.net)
|
|
self.la.available_wifis = self.wlans
|
|
self.la.start()
|
|
config = self.la.http_request.get_content_json()
|
|
print("start webserver end:", config)
|
|
return config
|
|
|
|
def save_config(self, config):
|
|
config = {
|
|
"ssid": config['ssid'],
|
|
"password": config['password'],
|
|
"pin": config['pin']
|
|
}
|
|
print("Save config:", config)
|
|
with open("/gs/config/initial_config.py", "w") as f:
|
|
f.write("config = " + ujson.dumps(config))
|
|
|
|
def switch_to_client_mode(self):
|
|
print("Switch to client mode")
|
|
ap = network.WLAN(network.AP_IF)
|
|
ap.active(False)
|
|
wlan = network.WLAN(network.STA_IF)
|
|
wlan.active(True)
|
|
wlan.connect(self.selected_ssid, self.wlan_password)
|
|
while not wlan.isconnected():
|
|
time.sleep(1)
|
|
print("Connected to", self.selected_ssid)
|
|
|
|
def setup_pico(self):
|
|
self.scan_wlans()
|
|
self.start_ap_mode()
|
|
config = self.get_initial_config_webserver()
|
|
self.save_config(config)
|
|
# self.stop_ap_mode()
|
|
# self.switch_to_client_mode()
|
|
|