python練手指令碼-定時檢測無響應程序並重啟

程序 Python · 發表 2019-04-21 22:19:03

摘要：總有一些程式在windows平臺表現不穩定，動不動一段時間就無響應，但又不得不用，每次都是發現問題了手動重啟，現在寫個指令碼定時檢測程序是否正常，自動重啟。涉及知識點 schedule定時任務排程 os.popen執行程式並...

總有一些程式在windows平臺表現不穩定，動不動一段時間就無響應，但又不得不用，每次都是發現問題了手動重啟，現在寫個指令碼定時檢測程序是否正常，自動重啟。

涉及知識點

schedule定時任務排程
os.popen執行程式並讀取解析執行結果

程式碼分解

指令碼主入口

if __name__ == '__main__':
#每5秒執行檢查任務
schedule.every(5).seconds.do(check_job)
#此處固定寫法，意思是每秒鐘schedule看下是否有pending的任務，有就執行
while True:
schedule.run_pending()
time.sleep(1)

schedule的其它示例

import schedule
import time

def job(message='stuff'):
print("I'm working on:", message)

#每10分鐘
schedule.every(10).minutes.do(job)
#每小時
schedule.every().hour.do(job, message='things')
#每天10點30分
schedule.every().day.at("10:30").do(job)

while True:
schedule.run_pending()
time.sleep(1)

檢查無響應程序並重啟

def check_job():
process_name = "xx.exe"
not_respond_list = list_not_response(process_name)
if len(not_respond_list) <= 0:
return
pid_params = " ".join(["/PID " + pid for pid in not_respond_list])
os.popen("taskkill /F " + pid_params)
if len(list_process(process_name)) <= 0:
start_program(r'E:\xx\xx.exe')
｝

查詢符合條件的程序列表

def list_process(process_name, not_respond=False):
cmd = 'tasklist /FI "IMAGENAME eq %s"'
if not_respond:
cmd = cmd + ' /FI "STATUS eq Not Responding"'
output = os.popen(cmd % process_name)
return parse_output(output.read())

def list_not_response(process_name):
return list_process(process_name, True)

解析命令執行結果

def parse_output(output):
print(output)
pid_list = []
lines = output.strip().split("\n")
if len(lines) > 2:
for line in lines[2:]:
pid_list.append(line.split()[1])
return pid_list

tasklist示例輸出

映像名稱PID 會話名會話#記憶體使用
========================= ======== ================ =========== ============
WizChromeProcess.exe1620 Console132,572 K

完整程式碼

import os
import time

import schedule


def parse_output(output):
print(output)
pid_list = []
lines = output.strip().split("\n")
if len(lines) > 2:
for line in lines[2:]:
pid_list.append(line.split()[1])
return pid_list


def list_not_response(process_name):
return list_process(process_name, True)


def list_process(process_name, not_respond=False):
cmd = 'tasklist /FI "IMAGENAME eq %s"'
if not_respond:
cmd = cmd + ' /FI "STATUS eq Not Responding"'
output = os.popen(cmd % process_name)
return parse_output(output.read())


def start_program(program):
os.popen(program)


def check_job():
process_name = "xx.exe"
not_respond_list = list_not_response(process_name)
if len(not_respond_list) <= 0:
return
pid_params = " ".join(["/PID " + pid for pid in not_respond_list])
os.popen("taskkill /F " + pid_params)
if len(list_process(process_name)) <= 0:
start_program(r'E:\xxx\xx.exe')


if __name__ == '__main__':
schedule.every(5).seconds.do(check_job)
while True:
schedule.run_pending()
time.sleep(1)