Python测试进阶——（3）编写Python程序监控计算机的服务是否正常运行

用python写了个简单的监控进程的脚本，当发现进程消失的时候，立即调用服务，开启服务。

脚本的工作原理是这样的：脚本读取配置文件，读取预先配置好的调用系统服务的路径和所要监控的服务在进程管理器中的进程名，之所以要用配置文件，是为了方便给需要的朋友，你只需要修改进程名和系统路径，源代码就不需要修改了。具体的看代码中的注释吧。下面的是配置文件 config.ini

[MonitorProgramPath]
ProgramPath: D:\Program Files (x86)\Tencent\WeChat\WeChat.exe

[MonitorProcessName]
ProcessName: WeChat.exe

上面可以根据你的需求配置不同的路径和进程名，我是需要监控 WeChat.exe 这个程序，那就配置好他的调用的系统路径和他在任务管理器里面的进程名。

下面来看看代码：

# -*- encoding: utf-8 -*-
import psutil
import os
import time
from configparser import ConfigParser

CONFIGFILE = 'config.ini'
config = ConfigParser()
config.read(CONFIGFILE)
ProgramPath = config.get('MonitorProgramPath', 'ProgramPath')
ProcessName = config.get('MonitorProcessName', 'ProcessName')
# 读取配置文件中的进程名和系统路径，这2个参数都可以在配置文件中修改
ProList = []
# 定义一个列表
pids = psutil.pids()

def main():
    for pid in pids:
        ProList.append(str(psutil.Process(pid).name()))
    # 把所有任务管理器中的进程名添加到列表
    if ProcessName in ProList:
        # 判断进程名是否在列表中，如果是True，则所监控的服务正在 运行状态，
        # 打印服务正常运行
        print('')
        print("Server is running...")
        print('')
    else:
        # 如果进程名不在列表中，即监控的服务挂了，则在log文件下记录日志
        # 日志文件名是以年月日为文件名

        print("Server is not running...")
        # f = open('.\\log\\' + time.strftime("%Y%m%d", time.localtime()) + '-exception.txt', 'a')
        f = open('log.txt', 'a')
        print('Server is not running,Begining to Restart Server...')
        # 打印服务状态
        f.write('\n' + 'Server is not running,Begining to Restart Server...' + '\n')
        f.write(time.strftime('%Y-%m-%d %H:%M:%S --%A--%c', time.localtime()) + '\n')
        # 写入时间和服务状态到日志文件中
        os.startfile(ProgramPath)
        # 调用服务重启
        f.write('Restart Server Success...' + '\n')
        f.write(time.strftime('%Y-%m-%d %H:%M:%S --%A--%c', time.localtime()))
        f.close()
        # 关闭文件
        print('Restart Server Success...')
        print(time.strftime('%Y-%m-%d %H:%M:%S --%A--%c', time.localtime()))
    del ProList[:]

# 清空列表，否则列表会不停的添加进程名，会占用系统资源

if __name__ == "__main__":
    while True:
        main()
        time.sleep(10)
        # 每隔10秒调用脚本看下服务是否正常，如果不正常则重启服务，如果正常，则打印服务正常

执行后，控制台返回如下信息：

因为没有打开WeChat程序，脚本每隔10s试图启动该进程，脚本正常工作。

打开log.txt，显示如下内容：

Server is not running,Begining to Restart Server...
-- :: --Friday--Fri Jun  ::
Restart Server Success...
-- :: --Friday--Fri Jun  ::
Server is not running,Begining to Restart Server...
-- :: --Friday--Fri Jun  ::
Restart Server Success...
-- :: --Friday--Fri Jun  ::
Server is not running,Begining to Restart Server...
-- :: --Friday--Fri Jun  ::
Restart Server Success...
-- :: --Friday--Fri Jun  ::