原理:
监控一个指定进程,每隔5秒钟获取其CPU、内存使用量超过60%即kill掉该进程,获取其句柄数,超过300也kill掉该进程
运行环境是windows 64位系统+python 2.7 64位 ,这里需要使用到psutil 类库,要另外安装。脚本里面可以自动安装,前提是你已经下载好了安装包psutil-3.3.0.win-amd64-py2.7.exe
下面看代码:
#!/usr/bin/env python
# -*- coding:utf-8 -*-
import time
from datetime import date, datetime, timedelta
import platform
import os
import win32ui,win32api,win32con,win32gui
import subprocess
def install():
print("install psutil...")
sysstr = platform.system()
if(sysstr =="Windows"):
print ("Call Windows tasks")
bit,type=platform.architecture()
print ("os bit: %s " % bit)
#print ("os type: %s " % type)
if(bit == "64bit"):
fileName="psutil-3.3.0.win-amd64-py2.7.exe";
else:
fileName="psutil-3.3.0.win32-py2.7.exe";
print("will install the file [%s]" % fileName)
#启动程序--4种方法
#subprocess.Popen(fileName); #非阻塞
#subprocess.Popen(fileName).wati(); #阻塞
#os.system(fileName); #阻塞
#win32api.ShellExecute(0, 'open', fileName, '','',0)
label = 'Setup' #此处假设主窗口名为tt
hld = win32gui.FindWindow(None, label)
count=0
while (hld == 0 and count<20):
print("the setup is no running,will run it...")
count += 1
win32api.ShellExecute(0, 'open', fileName, '','',0)
print("sleep 1 seconds...")
time.sleep(0.5)
#wnd = win32ui.GetForegroundWindow()
#print wnd.GetWindowText()
hld = win32gui.FindWindow(None, label)
print("hld is %s" % hld)
pwin=win32ui.FindWindow(None,label)
print("pwin is %s" % pwin)
print pwin.GetWindowText()
print("click...")
button2=win32ui.FindWindowEx(pwin,None,None,'下一步(&N) >') #找到按钮
button2.SendMessage(win32con.BM_CLICK, 0,-1)
button2=win32ui.FindWindowEx(pwin,None,None,'下一步(&N) >') #找到按钮
button2.SendMessage(win32con.BM_CLICK, 0,-1)
button2=win32ui.FindWindowEx(pwin,None,None,'下一步(&N) >') #找到按钮
button2.SendMessage(win32con.BM_CLICK, 0,-1)
button2=win32ui.FindWindowEx(pwin,None,None,'完成') #找到按钮
button2.SendMessage(win32con.BM_CLICK, 0,-1)
print("install done...")
# 鼠标点击
#print("click...")
#win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN,0,0)
#time.sleep(0.1)
#win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP,0,0)
#time.sleep(1)
#win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN,0,0)
#time.sleep(0.1)
#win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP,0,0)
#time.sleep(1)
#win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN,0,0)
#time.sleep(0.1)
#win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP,0,0)
#time.sleep(1)
#win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN,0,0)
#time.sleep(0.1)
#win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP,0,0)
elif(sysstr == "Linux"):
print ("Call Linux tasks")
else:
print ("Other System tasks")
try:
print("import psutil...")
import psutil
except Exception,e:
print Exception,":",e
install()
import psutil
def get_proc_by_id(pid):
return psutil.Process(pid)
def get_proc_by_name(pname):
""" get process by name
return the first process if there are more than one
"""
for proc in psutil.process_iter():
try:
# return if found one
if proc.name().lower() == pname.lower():
return proc
except psutil.AccessDenied:
pass
except psutil.NoSuchProcess:
pass
return None
def getProcess(pname, day=0, hour=0, min=0, second=0):
# Init time
now = datetime.now()
strnow = now.strftime('%Y-%m-%d %H:%M:%S')
print "now:",strnow
# First next run time
period = timedelta(days=day, hours=hour, minutes=min, seconds=second)
next_time = now + period
strnext_time = next_time.strftime('%Y-%m-%d %H:%M:%S')
print "next run time:",strnext_time
while True:
# Get system current time
iter_now = datetime.now()
iter_now_time = iter_now.strftime('%Y-%m-%d %H:%M:%S')
if str(iter_now_time) == str(strnext_time):
next_time = iter_now + period
strnext_time = next_time.strftime('%Y-%m-%d %H:%M:%S')
print "next run time:",strnext_time
try:
Process=get_proc_by_name(pname)
except Exception,e:
print Exception,":",e
if Process != None :
print "-------Found the process : %s" % Process.name();
print("pid is (%s)" % Process.pid);
Cpu_usage = Process.cpu_percent(interval=1)
print("cpu percent is (%s)" % Cpu_usage);
if (100-Cpu_usage) < 0.1 :
print "cpu percent larger 60,now will terminate this process !";
Process.terminate();
Process.wait(timeout=3);
continue
RAM_percent = Process.memory_percent()
print("memory percent is (%s)" % RAM_percent);
if (60-RAM_percent) < 0.1 :
print "memory percent larger 60,now will terminate this process !";
Process.terminate();
Process.wait(timeout=3);
continue
all_files = list(Process.open_files());
print("open files size is (%d)" % len(all_files));
if (len(all_files)>300) :
print "open files size larger 300,now will terminate this process !";
Process.terminate();
Process.wait(timeout=3);
continue
Threads_Num=Process.num_threads()
print("threads number is (%s)" % Threads_Num);
if (Threads_Num>200) :
print "threads number larger 200,now will terminate this process !";
Process.terminate();
Process.wait(timeout=3);
continue
else :
print "-------No found the process : %s" % pname;
continue
if __name__ == '__main__':
print("main....")
getProcess("QQ.exe",second=5)