进程占用CPU检测工具

原创

已于 2024-02-28 17:25:52 修改 · 337 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

于 2024-02-28 13:41:13 首次发布

#!/usr/bin/env python

import functools
import os
import sys
import subprocess
import psutil
import datetime
import time
import argparse

def getParam():
    parser = argparse.ArgumentParser(description="check cpu tool pararms")
    parser.add_argument('-n', '--node_name', required=True,
                        default='root')
    parser.add_argument('-p', '--node_pid', required=True, default=0)
    parser.add_argument('-s', '--save_file', default=True)
    args = parser.parse_args()
    return args

class Node:
    def __init__(self, name, pid):
        self.name = name
        self.proc = psutil.Process(pid)

    def get_values(self):
        self.cpu = self.proc.cpu_percent()
        self.mem = self.proc.memory_info().rss
        return self.cpu, self.mem

    def alive(self):
        return self.proc.is_running()


class CSVWriter:
    def __init__(self, filename, source_list):
        self.csv_path = os.path.join(os.path.dirname(
            os.p