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

# 啟動(dòng)服務(wù)
def start_service(service_name):
    if is_admin():
        try:
            cmd = 'NET START {}'.format(service_name)
            os.popen(cmd)
            logging.warning(f"成功：啟動(dòng)了服務(wù){(diào)service_name}")
        except Exception as e:
                logging.error(f"失?。簡?dòng)服務(wù){(diào)service_name}失敗, 錯(cuò)誤信息: {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)


# 獲取掉線的進(jìn)程完整執(zhí)行路徑
def find_process_by_name(process_names): 
    all__process = set()
    """獲取所有進(jìn)程的完成路徑"""  
    for proc in psutil.process_iter(['pid', 'name','exe']):
        all__process.add(proc.info['exe']) 
    """ 找出掉線的監(jiān)控進(jì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ù)是否啟動(dòng)
        get_service_status(service_anmes)
        time.sleep(60)

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

    loggingstars()

    logging.warning("============== 進(jìn)程監(jiān)控程序啟動(dòng) ============")

    # 需要監(jiān)測的進(jìn)程，為什么使用可執(zhí)行文件，是因?yàn)榘l(fā)現(xiàn)有進(jìn)程名會(huì)相同
    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)