#!/bin/python3
# -*- coding: utf-8 -*-

import os, sys
import time
import subprocess 
import logging
import configparser


config = {}

def process_detect():
    '''检测当前是否已经有在运行该程序，有则退出'''
    
    status, output = subprocess.getstatusoutput("ps -c | grep %s | grep -v grep | wc -l" % os.path.basename(__file__))
    print("Number of processes: %s" % output)

    process_sum = int(output)
    if process_sum > 1:
        print("Process already running, exit!");
        sys.exit()
    else:
        logging.info("Start ===== ")

def ping(addr):
    '''ping 指定ip，网络正常返回 True'''
    
    ret = os.system("ping -c 1 -W 1 %s > /dev/null" % addr)
    if ret == 0:
        return True
    else:
        return False

def power_220v_is_ok():
    '''ping 指定ip，判断220v市电是否正常'''
    
    if ping(config['220v_ip1']) or ping(config['220v_ip2']):
        return True
    else:
        return False

def network_is_ok():
    '''ping 指定外网IP 判断网络是否正常'''
    
    if ping(config['network_ip1']) or ping(config['network_ip2']):
        return True
    else:
        return False
        
def openwrt_reset():
    '''重启 openwrt'''
    logging.info("openwrt_reset!")
    os.system(config['openwrt_off'])
    time.sleep(2)
    os.system(config['openwrt_on'])
    time.sleep(config['openwrt_wait'])

def log_config():
    '''日志配置'''
    
    logging.basicConfig(filename=os.path.dirname(os.path.realpath(__file__)) + '/log.txt', level=logging.DEBUG, format="%(asctime)s %(message)s", datefmt="%Y/%m/%d %H:%M:%S -> ")
    # logging.basicConfig(level=logging.DEBUG, format="%(asctime)s %(message)s", datefmt="%Y-%m-%d %H:%M:%S ->")

def readConf():
    '''读取配置文件'''

    conf = configparser.ConfigParser()
    file_path = os.path.dirname(os.path.realpath(__file__)) + '/config.ini'     # 配置文件绝对路径
    print(file_path)
    conf.read(file_path, encoding = "utf-8")  

    config['220v_ip1'] = conf.get("power_220v", "ip1") 
    config['220v_ip2'] = conf.get("power_220v", "ip2") 
    config['220v_ping_sum'] = int(conf.get("power_220v", "ping_sum"))
    
    config['network_ip1'] = conf.get("network", "ip1") 
    config['network_ip2'] = conf.get("network", "ip2") 
    config['network_ping_sum'] = int(conf.get("network", "ping_sum"))
    config['network_reset_esxi'] = int(conf.get("network", "reset_esxi"))
    
    config['openwrt_off'] = conf.get("openwrt", "off_cmd") 
    config['openwrt_on'] = conf.get("openwrt", "on_cmd") 
    config['openwrt_wait'] = int(conf.get("openwrt", "wait"))
    
    config['sleep_sec'] = int(conf.get("default", "sleep_sec"))
    
    print("===== Configuration =====")
    print(config)
    print("====================")
     
def main():
    log_config()
    readConf()
    
    process_detect()
    
    power_220v_cnt = 0
    openwrt_cnt = 0
    openwrt_reset_wan_cnt = 0
    while True:
        sleep_sec = config['sleep_sec']
        
        if power_220v_is_ok():
            print("Power 220V ok!")
            power_220v_cnt = 0
        else:
            sleep_sec -= 4
            power_220v_cnt += 1
            logging.info("[%d/%d] 220V Power maybe off,checking again after %d seconds!" % (power_220v_cnt, config['220v_ping_sum'], config['sleep_sec']))
            
            if power_220v_cnt >= config['220v_ping_sum']:
                logging.info("Poweroff!")
                os.system("poweroff")
                
                while True:                    
                    time.sleep(1)
                    
                    
        if network_is_ok():
            print("Network ok!")
            openwrt_cnt = 0
            openwrt_reset_wan_cnt = 0
        else:
            sleep_sec -= 4
            openwrt_cnt += 1
            logging.info("[%d/%d] Network maybe disconnected!(reboot %d/%d)" % (openwrt_cnt, config['network_ping_sum'], openwrt_reset_wan_cnt, config['network_reset_esxi']))
            
            if openwrt_cnt >= config['network_ping_sum']:
                openwrt_cnt = 0
                openwrt_reset_wan_cnt += 1
                
                if openwrt_reset_wan_cnt >= config['network_reset_esxi']:
                    if power_220v_is_ok():              #市电正常时才进行重启，否则配置的市电故障关机时间比重启时间长的话，会一直重启无法关机
                        logging.info("Reboot!")
                        os.system("reboot")
                        while True:
                            time.sleep(1)
                            
                openwrt_reset()
        
        
        time.sleep(sleep_sec)


if __name__ == '__main__':  
    main()