【性能测试】获取性能系统指标之示例Python代码

#!/usr/bin/env python
#-*- coding: utf-8 -*-
import sys
import datetime
import time
import psutil
from ctypes import *


"""
性能测试示例Python代码：
    1、获取系统性能指标（内存、cpu、GDI、IO）
    2、保存数据至日志文件 (./******.log)
    3、可同时监控多个程序（xiadan:hexin）
"""


class PerfMonitor(object):

    def __init__(self, pnames, monitor_time):
        self.pnames = pnames
        self.monitor_time = monitor_time
        self.perflog = "./"
        self.perfimg = "./"

    def getGDIcount(self,PID):
        PH = windll.kernel32.OpenProcess(5432, 0, PID)
        GDIcount = windll.user32.GetGuiResources(PH, 0)
        windll.kernel32.CloseHandle(PH)
        return GDIcount

    def time_wait(self,proc_name):
        # 增加隐形等待时间
        pid_list = []
        now_data = time.time()
        while time.time() < now_data + 35:
            pid_data = list(filter(lambda x: x.name() == proc_name, psutil.process_iter()))
            if pid_data != []:
                pid_list.append(list(pid_data)[0].pid)
                break
            else:
                time.sleep(0.1)
        return pid_list

    def get_pid(self,pnames):
        # 获取进程号列表
        pnames = map(lambda x : (x + ".exe"), pnames.split(":"))
        pid_lists = []
        for proc_name in pnames:
            pid = self.time_wait(proc_name)
            pid_lists.extend(pid)
        return pid_lists

    def write_logs(self,log_filepath,usagestr):
        # 数据写入log日志
        fp=open(log_filepath,"a")
        fp.write(usagestr)
        fp.close()

    def get_nowtime(self):
        # 获取现在时间
        now_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
        return now_time

    def get_log_filepaths(self, p, monitor, types, check_times):
        name = p.name()
        names = name.split(".")[0]
        current = check_times
        usagestr = current + "\t" + str(monitor) + "\n"
        log_filepath = self.perflog + types + "-" + names + "-" + str(p.pid) + ".log"
        self.write_logs(log_filepath,usagestr)

    def perfrun(self):
        try:
            monitorsecond = self.monitor_time * 60
            begintime = int(time.time())
            p_list = []
            for pid in self.get_pid(self.pnames):
                p_list.append((pid,psutil.Process(int(pid))))
            while((int(time.time()) - begintime) <= monitorsecond):
                for (pid, p) in p_list:
                    # 内存
                    mem_time = self.get_nowtime()
                    memusage = p.memory_info().rss/1024.0/1024.0

                    # CPU
                    cpu_time = self.get_nowtime()
                    cpuusage = p.cpu_percent()

                    # GDI
                    gdi_time = self.get_nowtime()
                    GDIcount = self.getGDIcount(pid)

                    # IO
                    iousage = p.io_counters()
                    read_count_time = self.get_nowtime()
                    read_count = iousage.read_count
                    write_count_time = self.get_nowtime()
                    write_count = iousage.write_count
                    read_bytes_time = self.get_nowtime()
                    read_bytes = iousage.read_bytes
                    write_bytes_time = self.get_nowtime()
                    write_bytes = iousage.write_bytes

                    # 写入日志
                    self.get_log_filepaths(p, memusage, "mem", mem_time)
                    self.get_log_filepaths(p, cpuusage, "cpu", cpu_time)
                    self.get_log_filepaths(p, GDIcount, "gdi", gdi_time)
                    self.get_log_filepaths(p, read_count, "io-read-count", read_count_time)
                    self.get_log_filepaths(p, write_count, "io-write-count", write_count_time )
                    self.get_log_filepaths(p, read_bytes, "io-read-bytes", read_bytes_time)
                    self.get_log_filepaths(p, write_bytes, "io-write-bytes", write_bytes_time)

                time.sleep(2)

            now = datetime.datetime.now().strftime("%y-%m-%d-%H-%M-%S")
        except Exception as e:
            print(e)

if __name__ == '__main__':
    if(len(sys.argv) == 3):
        pnames=sys.argv[1]
        monitor_time=int(sys.argv[2])
    else:
        print("""|====================================================================================|""")
        print("""|                              Usage Instructions                                    |""")
        print("""|====================================================================================|""")
        print("""|  usage              : python perfmonitor.py pnames monitor_time (minute)           |""")
        print("""|  example            : python perfmonitor.py xiadan:hexin 10                        |""")
        print("""|====================================================================================|""")
        pnames = "xiadan"
        monitor_time = 5
    perfm = PerfMonitor(pnames, monitor_time)
    perfm.perfrun()