linux top命令模拟实现

import psutil
import time
from tabulate import tabulate

def get_process_info():
    processes = []
    for proc in psutil.process_iter(['pid', 'name', 'cpu_percent', 'memory_info']):
        try:
            proc_info = proc.as_dict(attrs=['pid', 'name', 'cpu_percent', 'memory_info'])
            proc_info['memory_usage'] = proc_info['memory_info'].rss / (1024 * 1024)  # MB
            processes.append(proc_info)
        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
            pass
    return processes

def display_top(processes, sort_by='cpu_percent'):
    headers = ["PID", "Name", f"{sort_by.capitalize()} (%)", "Memory Usage (MB)"]
    sorted_processes = sorted(processes, key=lambda x: x[sort_by], reverse=True)
    print(tabulate(sorted_processes, headers=headers, tablefmt="grid"))

def main():
    while True:
        processes = get_process_info()
        display_top(processes)
        time.sleep(1)

if __name__ == "__main__":
    main()