使用python脚本来监控进程

傷城~ 2023-10-16 22:49 33阅读 0赞

原理：

监控一个指定进程，每隔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)