import tkinter as tk
from tkinter import ttk
import psutil
 
# 運行此代碼前，請確保已經(jīng)安裝了 psutil 庫，可以使用 pip install psutil 進行安裝。
# 由于獲取進程信息可能會受到權限限制，某些進程的信息可能無法獲取，代碼中已經(jīng)對可能出現(xiàn)的異常進行了處理。
 
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 == '內(nèi)存占用率':
        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', '進程名稱', '內(nèi)存占用率')
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 *
 
 
# 列出系統(tǒng)當前所有進程。
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的回調(diào)函數(shù)
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()