python监控进程脚本

Python  /  管理员 发布于 7年前   144

本文实例为大家分享了python监控进程脚本的具体代码，供大家参考，具体内容如下

原理：

监控一个指定进程，每隔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: <span style="white-space:pre">   </span># return if found one    if proc.name().lower() == pname.lower():<span style="white-space:pre">   </span>     return proc<span style="white-space:pre">    </span>   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) 

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。