如何使用Python实现一个简单的window任务管理器

import tkinter as tk
from tkinter import ttk
import psutil
 
# 运行此代码前，请确保已经安装了 psutil 库，可以使用 pip install psutil 进行安装。
# 由于获取进程信息可能会受到权限限制，某些进程的信息可能无法获取，代码中已经对可能出现的异常进行了处理。
 
def get_process_info():
    process_list = []
    for proc in psutil.process_iter(['pid', 'name', 'memory_percent']):
        try:
            pid = proc.info['pid']
            name = proc.info['name']
            mem_percent = proc.info['memory_percent']
            process_list.append((pid, name, f'{mem_percent:.2f}%'))
        except (psutil.nosuchprocess, psutil.accessdenied, psutil.zombieprocess):
            continue
    return process_list
 
 
def populate_table():
    for item in process_table.get_children():
        process_table.delete(item)
    processes = get_process_info()
    for pid, name, mem_percent in processes:
        mem_value = float(mem_percent.strip('%'))
        tag = ""
        if mem_value > 10:
            tag = "high_mem"
        elif mem_value > 5:
            tag = "medium_mem"
        process_table.insert('', 'end', values=(pid, name, mem_percent), tags=(tag,))
 
    # 设置标签样式
    process_table.tag_configure("high_mem", foreground="red")
    process_table.tag_configure("medium_mem", foreground="green")
 
 
def sort_table(column, reverse):
    data = [(process_table.set(item, column), item) for item in process_table.get_children('')]
    if column == 'pid':
        data.sort(key=lambda t: int(t[0]), reverse=reverse)
    elif column == '内存占用率':
        data.sort(key=lambda t: float(t[0].strip('%')), reverse=reverse)
    else:
        data.sort(key=lambda t: t[0], reverse=reverse)
 
    for index, (_, item) in enumerate(data):
        process_table.move(item, '', index)
 
    process_table.heading(column, command=lambda: sort_table(column, not reverse))
 
 
root = tk.tk()
root.title("任务管理器")
root.geometry("700x500")
root.configure(bg="#f4f4f9")
 
style = ttk.style()
style.theme_use('clam')
style.configure('treeview', background="#e9e9f3", foreground="#333", fieldbackground="#e9e9f3",
                rowheight=25, font=('segoe ui', 10))
style.map('treeview', background=[('selected', '#73a6ff')])
style.configure('treeview.heading', background="#d1d1e0", foreground="#333", font=('segoe ui', 10, 'bold'))
 
columns = ('pid', '进程名称', '内存占用率')
process_table = ttk.treeview(root, columns=columns, show='headings')
for col in columns:
    process_table.heading(col, text=col, command=lambda c=col: sort_table(c, false))
    process_table.column(col, width=200, anchor='center')
process_table.pack(pady=20, padx=20, fill=tk.both, expand=true)
 
populate_table()
 
refresh_button = ttk.button(root, text="刷新", command=populate_table)
refresh_button.pack(pady=10)
 
root.mainloop()

import os
# import win32api
import win32gui
import win32process
from ctypes import *
 
 
# 列出系统当前所有进程。
def getprocesslist():
    os.system("tasklist")
 
 
# 结构体
class threadentry32(structure):
    _fields_ = [('dwsize', c_ulong),
                ('cntusage', c_ulong),
                ('th32threadid', c_ulong),
                ('th32ownerprocessid', c_ulong),
                ('tpbasepri', c_long),
                ('tpdeltapri', c_long),
                ('dwflags', c_ulong)]
 
 
# 获取指定进程的所有线程
def getthreadofprocess(pid):
    dll = windll.loadlibrary("kernel32.dll")
    snapshothandle = dll.createtoolhelp32snapshot(0x00000004, pid)
    struct = threadentry32()
    struct.dwsize = sizeof(threadentry32)
    flag = dll.thread32first(snapshothandle, byref(struct))
 
    while flag != 0:
        if(struct.th32ownerprocessid == int(pid)):
            print("线程id："+str(struct.th32threadid))
        flag = dll.thread32next(snapshothandle, byref(struct))
    dll.closehandle(snapshothandle)
 
 
# enumwindows的回调函数
def callback(hwnd, windows):
    pidlist = win32process.getwindowthreadprocessid(hwnd)
    for pid in pidlist:
        windows.setdefault(pid, [])
        windows[pid].append(hwnd)
 
 
# 显示和隐藏指定进程的窗口
def changewindowstate(pid, status):
    windows = {}
    win32gui.enumwindows(callback, windows)
    try:
        hwndlist = windows[int(pid)]
        # 显示/隐藏窗口
        for hwnd in hwndlist:
            win32gui.showwindow(hwnd, int(status))
    except:
        print("进程不存在")
 
 
# 强行结束指定进程
def killprocess(pid):
    cmd = 'taskkill /pid ' + pid + ' /f'
    try:
        os.system(cmd)
    except exception as e:
        print(e)
 
 
if __name__ == "__main__":
    while(true):
        print()
        print()
        print("************************************")
        print("*                                  *")
        print("*     进程管理器                   *")
        print("*                                  *")
        print("*     1.获取所有进程               *")
        print("*     2.获取指定进程的所有线程     *")
        print("*     3.显示和隐藏指定进程的窗口   *")
        print("*     4.强行结束指定进程           *")
        print("*                                  *")
        print("************************************")
        option = input("请选择功能：")
 
        if option == "1":
            getprocesslist()
        elif option == "2":
            pid = input("请输入进程的pid：")
            getthreadofprocess(pid)
        elif option == "3":
            pid = input("请输入进程的pid：")
            status = input("隐藏输入0，显示输入1：")
            changewindowstate(pid, status)
        elif option == "4":
            pid = input("请输入进程的pid：")
            killprocess(pid)
        else:
            exit()