main.py

#!/usr/bin/env python
# coding:utf8
# sky

from module import logger, common, stats,  config
from core import install,stat_dump, health
import gevent
from gevent import monkey
import sys

import os, time

def Install():
    log.log("info", "安装程序启动")
    install.soft_install()

def Status():
    log.log("info", "监控程序启动")
    stat_dump.dump()

def Health():
    log.log("info", "检查程序启动")
    health.check()
    
if __name__ == "__main__":
    monkey.patch_all() 
    if len(sys.argv)==2:
        action=sys.argv[1]
    if action=="restart":
        force=1
    elif action=="start":
        force=0

    base_dir=common.base_dir(__file__)
    os.chdir(base_dir)
    #os.chroot(base_dir)

    # 日志
    logger_config="conf/logger.yml"
    log=logger.logger(logger_config)

    # 获取配置
    conf_config="conf/monitor.yml"
    conf=config.config(conf_config)
    conf_res=conf.get_monitor_conf()
    pid_file=conf_res["pid"]
    log.log("info", "main process: %s" % os.getpid())
    
    # 生成pid文件
    try:
        if os.path.exists(pid_file):
            with open(pid_file, "r", encoding="utf8") as f:
                pid=int(f.read())
            pid_obj=stats.soft_status(pid)
            if pid_obj.pid_exist():
                if force==0:
                    log.log("info", "监控程序(Pid: %s)正在运行, 请不要重复启动" % pid)
                    exit()
                elif force==1:
                    os.kill(pid, 9)
            with open(pid_file, "w", encoding="utf8") as f:
                pid=str(os.getpid())
                f.write(pid)
        else:
            with open(pid_file, "w+", encoding="utf8") as f:
                pid=str(os.getpid())
                f.write(pid)

    except Exception as e:
        log.log("error", "无法生成pid文件:", e)
        exit()
    
    # 启动本地安装和监控程序
    try:
        g1=gevent.spawn(Install, )
        g2=gevent.spawn(Status, )
        #g3=gevent.spawn(Health, )
        gevent_list=[g1, g2]
        gevent.joinall(gevent_list)
    except Exception as e:
        log.log("critical", "监控程序无法启动: %s" % e)