CO2_Ambient_and_ifttt.py

from m5stack import *
import machine
import gc
import utime
import uos
import _thread
import ntptime
from tiny_ifttt import tiny_ifttt


# 変数宣言
Am_err              = 1     # グローバル
Disp_mode           = 1     # グローバル
lcd_mute            = False # グローバル
data_mute           = False # グローバル
am_interval         = 60    # Ambientへデータを送るサイクル（秒）
co2_interval        = 5     # MH-19Bへco2測定値要求コマンドを送るサイクル（秒）
TIMEOUT             = 30    # 何らかの事情でCO2更新が止まった時のタイムアウト（秒）のデフォルト値
CO2_RED             = 1000  # co2濃度の換気閾値（ppm）のデフォルト値
AM_ID               = None
AM_WKEY             = None
IFTTT_KEY           = None
co2                 = 0


# @cinimlさんのファーム差分吸収ロジック
class AXPCompat(object):
    def __init__(self):
        if( hasattr(axp, 'setLDO2Vol') ):
            self.setLDO2Vol = axp.setLDO2Vol
        else:
            self.setLDO2Vol = axp.setLDO2Volt

axp = AXPCompat()


# 時計表示スレッド関数
def time_count ():
    global Disp_mode
    global Am_err

    while True:
        if Am_err == 0 : # Ambient通信不具合発生時は時計の文字が赤くなる
            fc = lcd.WHITE
        else :
            fc = lcd.RED

        if Disp_mode == 1 : # 表示回転処理
            lcd.rect(67, 0, 80, 160, lcd.BLACK, lcd.BLACK)
            lcd.font(lcd.FONT_DefaultSmall, rotate = 90)
            lcd.print('{}-{:02d}-{:02d} {:02d}:{:02d}:{:02d}'.format(*time.localtime()[:6]), 78, 40, fc)
        else :
            lcd.rect(0 , 0, 13, 160, lcd.BLACK, lcd.BLACK)
            lcd.font(lcd.FONT_DefaultSmall, rotate = 270)
            lcd.print('{}-{:02d}-{:02d} {:02d}:{:02d}:{:02d}'.format(*time.localtime()[:6]), 2, 125, fc)

        utime.sleep(1)

def fan_control ():
    global co2

    my_key = IFTTT_KEY

    # 初期化
    ti= tiny_ifttt(my_key, debug=True)

    state = False
    while True:
        # フラグONかつ500未満で換気扇OFF
        if co2 < 500:
            if state == True:
                ti.trigger('Fan_control_off', value1='', value2='', value3='')
                state = False

        # フラグOFFかつ1400以上で換気扇ON
        if co2 >= 1400:
            if state == False:
                ti.trigger('Fan_control_on', value1='', value2='', value3='')
                state = True

# 表示OFFボタン処理スレッド関数
def buttonA_wasPressed():
    global lcd_mute

    if lcd_mute :
        lcd_mute = False
    else :
        lcd_mute = True

    if lcd_mute == True :
        axp.setLDO2Vol(0)   #バックライト輝度調整（OFF）
    else :
        axp.setLDO2Vol(2.7) #バックライト輝度調整（中くらい）


# 表示切替ボタン処理スレッド関数
def buttonB_wasPressed():
    global Disp_mode

    if Disp_mode == 1 :
        Disp_mode = 0
    else :
        Disp_mode = 1

    draw_lcd()


# 表示モード切替時の枠描画処理関数
def draw_lcd():
    global Disp_mode

    lcd.clear()

    if Disp_mode == 1 :
        lcd.line(66, 0, 66, 160, lcd.LIGHTGREY)
    else :
        lcd.line(14, 0, 14, 160, lcd.LIGHTGREY)

    draw_co2()


# CO2値表示処理関数
def draw_co2():
    global Disp_mode
    global lcd_mute
    global data_mute
    global CO2_RED
    global co2

    if data_mute or (co2 == 0) : # タイムアウトで表示ミュートされてるか、初期値のままならco2値非表示（黒文字化）
        fc = lcd.BLACK
    else :
        if co2 >= CO2_RED :  # CO2濃度閾値超え時は文字が赤くなる
            fc = lcd.RED
            if lcd_mute == True :   # CO2濃度閾値超え時はLCD ON
                axp.setLDO2Vol(2.7) # バックライト輝度調整（中くらい）
        else :
            fc = lcd.WHITE
            if lcd_mute == True :
                axp.setLDO2Vol(0)   # バックライト輝度調整（中くらい）

    if Disp_mode == 1 : # 表示回転処理
        lcd.rect(0, 0, 65, 160, lcd.BLACK, lcd.BLACK)
        lcd.font(lcd.FONT_DejaVu18, rotate = 90) # 単位(ppm)の表示
        lcd.print('ppm', 37, 105, fc)
        lcd.font(lcd.FONT_DejaVu24, rotate = 90) # co2値の表示
        lcd.print(str(co2), 40, 125 - (len(str(co2))* 24), fc)
    else :
        lcd.rect(15 , 0, 80, 160, lcd.BLACK, lcd.BLACK)
        lcd.font(lcd.FONT_DejaVu18, rotate = 270) # 単位(ppm)の表示
        lcd.print('ppm', 43, 55, fc)
        lcd.font(lcd.FONT_DejaVu24, rotate = 270) # co2値の表示
        lcd.print(str(co2), 40, 35 + (len(str(co2))* 24), fc)


# MH-Z19Bデータのチェックサム確認関数
def checksum_chk(data):
    sum = 0
    for a in data[1:8]:
        sum = (sum + a) & 0xff
    c_sum = 0xff - sum + 1
    if c_sum == data[8]:
        return True
    else:
        print("c_sum un match!!")
        return False


# co2_set.txtの存在/中身チェック関数
def co2_set_filechk():
    global CO2_RED
    global TIMEOUT
    global AM_ID
    global AM_WKEY

    scanfile_flg = False
    for file_name in uos.listdir('/flash') :
        if file_name == 'co2_set.txt' :
            scanfile_flg = True

    if scanfile_flg :
        print('>> found [co2_set.txt] !')
        with open('/flash/co2_set.txt' , 'r') as f :
            for file_line in f :
                filetxt = file_line.strip().split(':')
                if filetxt[0] == 'CO2_RED' :
                    if int(filetxt[1]) >= 1 :
                        CO2_RED = int(filetxt[1])
                        print('- CO2_RED: ' + str(CO2_RED))
                elif filetxt[0] == 'TIMEOUT' :
                    if int(filetxt[1]) >= 1 :
                        TIMEOUT = int(filetxt[1])
                        print('- TIMEOUT: ' + str(TIMEOUT))
                elif filetxt[0] == 'AM_ID' :
                    AM_ID = str(filetxt[1])
                    print('- AM_ID: ' + str(AM_ID))
                elif filetxt[0] == 'AM_WKEY' :
                    if len(filetxt[1]) == 16 :
                        AM_WKEY = str(filetxt[1])
                        print('- AM_WKEY: ' + str(AM_WKEY))
                elif filetxt[0] == 'IFTTT_KEY' :
                    IFTTT_KEY = str(filetxt[1])
                    print('- IFTTT_KEY: ' + str(IFTTT_KEY))
    else :
        print('>> no [co2_set.txt] !')
    return scanfile_flg


# メインプログラムはここから（この上はプログラム内関数）


# WiFi設定
import wifiCfg
wifiCfg.autoConnect(lcdShow=True)


# 画面初期化
axp.setLDO2Vol(2.7) #バックライト輝度調整（中くらい）
draw_lcd()


# MH-19B UART設定
mhz19b = machine.UART(1, tx=0, rx=26)
mhz19b.init(9600, bits=8, parity=None, stop=1)


# ユーザー設定ファイル読み込み
co2_set_filechk()


# Ambient設定
if (AM_ID is not None) and (AM_WKEY is not None) : # Ambient設定情報があった場合
    import ambient
    am_co2 = ambient.Ambient(AM_ID, AM_WKEY)


# RTC設定
utime.localtime(ntptime.settime())


# 時刻表示スレッド起動
_thread.start_new_thread(time_count , ())

# IFTTT 要求スレッド起動
_thread.start_new_thread(fan_control, ())

# ボタン検出スレッド起動
btnA.wasPressed(buttonA_wasPressed)
btnB.wasPressed(buttonB_wasPressed)


# タイムカウンタ初期値設定
co2_tc = utime.time()
am_tc = utime.time()


# メインルーチン
while True:
    if (utime.time() - co2_tc) >= co2_interval : # co2要求コマンド送信
        mhz19b_data = bytearray(9)
        mhz19b.write(b'\xff\x01\x86\x00\x00\x00\x00\x00\x79')   # co2測定値リクエスト
        utime.sleep(0.1)
        mhz19b.readinto(mhz19b_data, len(mhz19b_data))
        # co2測定値リクエストの応答
        if mhz19b_data[0] == 0xff and mhz19b_data[1] == 0x86 and checksum_chk(mhz19b_data) == True :    # 応答かどうかの判定とチェックサムチェック
            co2_tc = utime.time()
            co2 = mhz19b_data[2] * 256 + mhz19b_data[3]
            data_mute = False
            draw_co2()
            print(str(co2) + ' ppm / ' + str(co2_tc))
            if (AM_ID is not None) and (AM_WKEY is not None) :  # Ambient設定情報があった場合
                if (utime.time() - am_tc) >= am_interval :      # インターバル値の間隔でAmbientへsendする
                    am_tc = utime.time()
                    try :                                       # ネットワーク不通発生などで例外エラー終了されない様に try except しとく
                        r = am_co2.send({'d1': co2})
                        print('Ambient send OK! / ' + str(r.status_code) + ' / ' + str(Am_err))
                        Am_err = 0
                        am_tc = utime.time()
                        r.close()
                    except:
                        print('Ambient send ERR! / ' + str(Am_err))
                        Am_err = Am_err + 1
        utime.sleep(1)

    if (utime.time() - co2_tc) >= TIMEOUT : # co2応答が一定時間無い場合はCO2値表示のみオフ
        data_mute = True
        draw_co2()

    utime.sleep(0.1)
    gc.collect()