import psutil  
import time
import subprocess
import logging
import logging.handlers
import wmi
import ctypes
import os

# 獲取服務(wù)狀態(tài)
def get_service_status(service_names):
    wmiobj = wmi.WMI()
    for service_name in service_names:
        services = wmiobj.Win32_Service(Name = service_name)
        if services:
            # 服務(wù)狀態(tài)是停止就開啟服務(wù)
            if services[0].state == "Stopped":
                start_service(service_name)
            # return services[0].state
        else:
            logging.error(f"失敗：服務(wù){(diào)service_name}不存在")

#判斷是否有管理員權(quán)限
def is_admin():
    try:
        return ctypes.windll.shell32.IsUserAnAdmin()
    except:
        return False

# 啟動服務(wù)
def start_service(service_name):
    if is_admin():
        try:
            cmd = 'NET START {}'.format(service_name)
            os.popen(cmd)
            logging.warning(f"成功：啟動了服務(wù){(diào)service_name}")
        except Exception as e:
                logging.error(f"失?。簡臃?wù){(diào)service_name}失敗, 錯誤信息: {e}")
    else:
        logging.warning(f"不是admin權(quán)限")

# 日志配置
def loggingstars():
    logger = logging.getLogger()
    logger.setLevel(logging.DEBUG)

    handler = logging.StreamHandler()
    logger.addHandler(handler)

    file_handler = logging.handlers.RotatingFileHandler(filename='process_name.log', maxBytes=1024*1024*90, backupCount=10)
    file_handler.setLevel(logging.WARNING)

    formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
    file_handler.setFormatter(formatter)

    logger.addHandler(file_handler)


# 獲取掉線的進程完整執(zhí)行路徑
def find_process_by_name(process_names): 
    all__process = set()
    """獲取所有進程的完成路徑"""  
    for proc in psutil.process_iter(['pid', 'name','exe']):
        all__process.add(proc.info['exe']) 
    """ 找出掉線的監(jiān)控進程的完整路徑 """
    new_list = [item for item in process_names if item not in all__process]
    return new_list


def start_process(process_names, service_anmes):
    while True:
        # 檢查服務(wù)是否啟動
        get_service_status(service_anmes)
        time.sleep(60)

        stop_process = find_process_by_name(process_names)
        
        if not stop_process:
            continue
        """ 啟動掉線的進程 """
        for app in stop_process:
            try:
                subprocess.Popen(f"{app}")
                logging.warning(f"成功：啟動了進程{app}")
            except Exception as e:
                logging.error(f"失?。簡舆M程{app}失敗, 錯誤信息: {e}")
        # 如果需要啟動進程的話，暫停20分鐘
        time.sleep(20*60)
  
if __name__ == "__main__":

    loggingstars()

    logging.warning("============== 進程監(jiān)控程序啟動 ============")

    # 需要監(jiān)測的進程，為什么使用可執(zhí)行文件，是因為發(fā)現(xiàn)有進程名會相同
    process_names = [
        "C:\\UserApp\\app\\baidu-translate-client\\百度翻譯.exe",
        "C:\\UserApp\\app\\dingding\\main\\current\\DingTalk.exe"
    ]

   # 需要監(jiān)測的服務(wù)名
    service_anmes = ["MySQL80", "JetBrainsEtwHost"]
    start_process(process_names, service_anmes)