esp01_update.py

import network, time, machine, ubinascii, os
import usocket
from machine import Pin, PWM, Timer, WDT
from umqtt.simple import MQTTClient

class debug:
    def print(msg="",msg2=""):
        #print(msg,msg2)
        pass

class LED():
    def __init__(self,pin,pwm=True):
        self.pin = pin
        self.strong=2
        self.state = False
        self.timer = None
        self.isBlink = False
        if pwm:
            self.pwm = PWM(Pin(pin))
            self.pwm.freq(1024)
            self.pwm.duty(0)
        self.off()
        time.sleep(0.01)
        self.off()
        time.sleep(0.01)
        self.off()
        time.sleep(0.01)
            
    def on(self,strong=-1):
        self.state = True
        if not strong == -1:
            self.strong= strong
        self.pwm.duty(self.strong)
        
    def off(self):
        self.state = False
        self.pwm.duty(0)
    
    def run(self,n):
        if self.isBlink:
            if self.state:
                self.off()
            else:
                self.on()
    
    def blink(self,peroid):
        _p_ = int(peroid*1000)
        if _p_ == 0:
            self.isBlink = False
            self.off()
            return
        else:
            self.isBlink = True
            self.on()
        
        if self.timer == None:
            self.timer = Timer(1)
            self.timer.init(period=_p_, mode=Timer.PERIODIC, callback=self.run)
            debug.print("LED:default use Timer(0)",self.timer)
        else:
            self.timer.deinit()
            self.timer = Timer(1)
            self.timer.init(period=_p_, mode=Timer.PERIODIC, callback=self.run)
            

class WiFi:

    def disconnect():
        WiFi.sta.disconnect()

    def check():
        print(WiFi.sta.isconnected())

    def onlilne(cb):
        WiFi.onlineCallback = cb

    def startKeepConnect():
        WiFi.timer = Timer(0)
        WiFi.timer.init(period=3000, mode=Timer.PERIODIC, callback=WiFi.checkConnection)
        
    def checkConnection(t):
        #print("wifi:",WiFi.sta.isconnected())
        if not WiFi.sta.isconnected():
            debug.print("connect broke ! retry...")
            WiFi.connect(WiFi.ssid,WiFi.pwd)
            debug.print("!!!! online callback... !!!!")
        
    def connect(ssid="ttt",pwd="webduino"):
        WiFi.ssid = ssid
        WiFi.pwd = pwd
        WiFi.sta = sta_if = network.WLAN(network.STA_IF)
        sta_if.active(True)
        debug.print('connecting to network...')
        WiFi.onlineCallback(False)
        if not sta_if.isconnected():
            try:
                sta_if.connect(WiFi.ssid,WiFi.pwd)
            except:
                debug.print("connect...")
        cnt = 0
        while not sta_if.isconnected():
            cnt = cnt + 1
            time.sleep(0.5)
            if cnt == 10:
                cnt = 0
                debug.print("retry connect...")
        WiFi.ip = WiFi.sta.ifconfig()[0];
        WiFi.onlineCallback(True)
        WiFi.startKeepConnect()
        return True

    def ip():
        print(Wifi.sta.ifconfig())


class MQTT():
    
    def connect(server = 'mqtt1.webduino.io',user ='webduino' ,pwd='webduino'):
        MQTT.server = server
        MQTT.user = user
        MQTT.pwd = pwd
        MQTT.client = MQTTClient('guest', server, user=user, password=pwd)
        state = True if MQTT.client.connect() == 0 else False
        try:
            MQTT.subTopic
            debug.print("resubscribe:",MQTT.subTopic)
            MQTT.sub(MQTT.subTopic,MQTT.callback)
        except:
            pass
        return state
        
    def pub(topic,msg):
        MQTT.client.publish(topic,msg)

    def sub(topic,cb):
        MQTT.subTopic = topic
        MQTT.callback = cb
        MQTT.client.set_callback(cb)
        MQTT.client.subscribe(topic)

    def checkMsg():
        try:
            MQTT.client.check_msg()
        except:
            pass



class esp01():
    
    def setup(name='esp01'):
        esp01.led = LED(2)
        esp01.wifi = WiFi
        esp01.mqtt = MQTT
        esp01.name = name
        esp01.wifi.onlilne(esp01.online)
        
    def online(status):
        if status:
            debug.print("online,stop blink...")
            esp01.led.blink(1)
            esp01.mqtt.connect()
            esp01.led.blink(0)
        else:
            debug.print("offline,blink...")
            esp01.led.blink(2)
            pass

    def connect(ssid='webduino.io',pwd='webduino'):
        while True:
            if esp01.wifi.connect(ssid,pwd):
                if esp01.mqtt.connect('mqtt1.webduino.io','webduino','webduino'):
                    break
        esp01.mqtt.sub(esp01.name+'/#',esp01.onMsg)       
        debug.print("WiFi Ready , MQTT Ready , ready to go...")

    def onMsg(topic,msg):
        print("esp01 onMsg:",topic,msg)
        
    def loop():
        while True:
            esp01.mqtt.checkMsg()
            time.sleep(0.1)

#print("start init..")
#esp01.setup("esp01")
#esp01.connect("KingKit_2.4G","webduino")
#print("Ready to Go....")
#esp01.loop()

def save(file):
    response = urequests.get('http://webduino.tw/'+file)
    f = open(file, 'w')
    f.write(response.text)
    f.close()
    print("OK.")