import os
import psutil
import ctypes
import sys
import tkinter as tk
from tkinter import ttk, messagebox, filedialog
import csv
import time
import threading
 
# 檢查并提升管理員權(quán)限
def run_as_admin():
    if sys.platform != 'win32':
        return False
 
    try:
        if ctypes.windll.shell32.IsUserAnAdmin():
            return True
    except:
        pass
 
    ctypes.windll.shell32.ShellExecuteW(None, "runas", sys.executable, " ".join(sys.argv), None, 1)
    sys.exit()
 
# 獲取所有進(jìn)程信息
def get_processes():
    processes = []
    for proc in psutil.process_iter(['pid', 'name', 'memory_info', 'exe']):
        try:
            processes.append((proc.info['pid'], proc.info['name'], proc.info['memory_info'].rss, proc.info['exe']))
        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
            continue
    return processes
 
# 計(jì)算所有進(jìn)程的內(nèi)存使用總和
def get_total_memory_usage():
    total_memory = 0
    for proc in psutil.process_iter(['memory_info']):
        try:
            total_memory += proc.info['memory_info'].rss
        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
            continue
    return total_memory
 
# 結(jié)束選中的進(jìn)程
def kill_process():
    selected_item = process_tree.selection()
    if not selected_item:
        messagebox.showwarning("警告", "請選擇一個(gè)進(jìn)程！")
        return
 
    pid = int(process_tree.item(selected_item, 'values')[0])
    process_name = process_tree.item(selected_item, 'values')[1]
 
    # 確認(rèn)對話框
    confirm = messagebox.askyesno("確認(rèn)", f"確定要結(jié)束進(jìn)程 {process_name} (PID: {pid}) 嗎？")
    if not confirm:
        return
 
    try:
        process = psutil.Process(pid)
        process.terminate()
        messagebox.showinfo("成功", f"進(jìn)程 {process_name} (PID: {pid}) 已結(jié)束！")
        refresh_processes()
    except Exception as e:
        messagebox.showerror("錯(cuò)誤", f"結(jié)束進(jìn)程時(shí)出錯(cuò): {e}")
 
# 刷新進(jìn)程列表
def refresh_processes():
    for row in process_tree.get_children():
        process_tree.delete(row)
    processes = get_processes()
    for proc in processes:
        process_tree.insert("", "end", values=proc)
 
    # 更新內(nèi)存使用信息
    update_memory_usage()
 
# 更新內(nèi)存使用信息
def update_memory_usage():
    total_memory = get_total_memory_usage()
    memory_before.set(f"清理前內(nèi)存使用: {total_memory / 1024 / 1024:.2f} MB")
    memory_after.set(f"清理后內(nèi)存使用: {total_memory / 1024 / 1024:.2f} MB")
 
# 內(nèi)存清理功能
def clear_memory():
    try:
        # 獲取清理前的內(nèi)存使用總和
        before_memory = get_total_memory_usage()
 
        # 釋放未使用的內(nèi)存
        ctypes.windll.psapi.EmptyWorkingSet(ctypes.c_ulong(-1))
 
        # 獲取清理后的內(nèi)存使用總和
        after_memory = get_total_memory_usage()
 
        # 更新界面顯示
        memory_before.set(f"清理前內(nèi)存使用: {before_memory / 1024 / 1024:.2f} MB")
        memory_after.set(f"清理后內(nèi)存使用: {after_memory / 1024 / 1024:.2f} MB")
        messagebox.showinfo("成功", "內(nèi)存清理完成！")
    except Exception as e:
        messagebox.showerror("錯(cuò)誤", f"清理內(nèi)存時(shí)出錯(cuò): {e}")
 
# 導(dǎo)出進(jìn)程列表為 CSV 文件
def export_processes():
    file_path = filedialog.asksaveasfilename(
        defaultextension=".csv",
        filetypes=[("CSV 文件", "*.csv"), ("所有文件", "*.*")],
        title="保存進(jìn)程列表"
    )
    if not file_path:
        return
 
    try:
        with open(file_path, mode="w", newline="", encoding="utf-8") as file:
            writer = csv.writer(file)
            # 寫入表頭
            writer.writerow(["PID", "進(jìn)程名", "內(nèi)存使用 (MB)", "文件位置"])
            # 寫入進(jìn)程數(shù)據(jù)
            for row in process_tree.get_children():
                values = process_tree.item(row, 'values')
                writer.writerow(values)
        messagebox.showinfo("成功", f"進(jìn)程列表已導(dǎo)出到 {file_path}")
    except Exception as e:
        messagebox.showerror("錯(cuò)誤", f"導(dǎo)出進(jìn)程列表時(shí)出錯(cuò): {e}")
 
# 排序函數(shù)
def treeview_sort_column(tv, col, reverse):
    l = [(tv.set(k, col), k) for k in tv.get_children('')]
    try:
        l.sort(key=lambda t: int(t[0]) if col == "memory" else t[0], reverse=reverse)
    except ValueError:
        l.sort(reverse=reverse)
 
    for index, (val, k) in enumerate(l):
        tv.move(k, '', index)
 
    tv.heading(col, command=lambda: treeview_sort_column(tv, col, not reverse))
 
# 監(jiān)控新進(jìn)程
def monitor_new_processes():
    global previous_processes
 
    while True:
        current_processes = get_processes()
        current_pids = {proc[0] for proc in current_processes}
        previous_pids = {proc[0] for proc in previous_processes}
 
        # 檢查新進(jìn)程
        new_pids = current_pids - previous_pids
        if new_pids:
            for proc in current_processes:
                if proc[0] in new_pids:
                    messagebox.showinfo(
                        "新進(jìn)程提醒",
                        f"新進(jìn)程已啟動(dòng):\n\n"
                        f"進(jìn)程名: {proc[1]}\n"
                        f"PID: {proc[0]}\n"
                        f"文件位置: {proc[3]}"
                    )
 
        previous_processes = current_processes
        time.sleep(5)  # 每 5 秒檢查一次
 
# 創(chuàng)建主窗口
def create_gui():
    global memory_before, memory_after, process_tree, previous_processes
 
    root = tk.Tk()
    root.title("內(nèi)存清理工具")
    root.geometry("800x600")
 
    # 初始化進(jìn)程列表
    previous_processes = get_processes()
 
    # 啟動(dòng)新進(jìn)程監(jiān)控線程
    monitor_thread = threading.Thread(target=monitor_new_processes, daemon=True)
    monitor_thread.start()
 
    # 標(biāo)題
    title_label = tk.Label(root, text="Windows 內(nèi)存清理工具", font=("Arial", 16))
    title_label.pack(pady=10)
 
    # 內(nèi)存信息顯示
    memory_before = tk.StringVar()
    memory_after = tk.StringVar()
 
    memory_before.set("清理前內(nèi)存使用: N/A")
    memory_after.set("清理后內(nèi)存使用: N/A")
 
    before_label = tk.Label(root, textvariable=memory_before, font=("Arial", 12))
    before_label.pack(pady=5)
 
    after_label = tk.Label(root, textvariable=memory_after, font=("Arial", 12))
    after_label.pack(pady=5)
 
    # 清理內(nèi)存按鈕
    clear_button = tk.Button(root, text="清理內(nèi)存", command=clear_memory, font=("Arial", 14), bg="lightblue")
    clear_button.pack(pady=10)
 
    # 進(jìn)程列表
    process_frame = tk.Frame(root)
    process_frame.pack(fill="both", expand=True, padx=10, pady=10)
 
    columns = ("pid", "name", "memory", "exe")
    process_tree = ttk.Treeview(process_frame, columns=columns, show="headings")
    process_tree.heading("pid", text="PID", command=lambda: treeview_sort_column(process_tree, "pid", False))
    process_tree.heading("name", text="進(jìn)程名", command=lambda: treeview_sort_column(process_tree, "name", False))
    process_tree.heading("memory", text="內(nèi)存使用 (MB)", command=lambda: treeview_sort_column(process_tree, "memory", False))
    process_tree.heading("exe", text="文件位置", command=lambda: treeview_sort_column(process_tree, "exe", False))
 
    process_tree.column("pid", width=100, anchor="center")
    process_tree.column("name", width=200, anchor="w")
    process_tree.column("memory", width=150, anchor="center")
    process_tree.column("exe", width=300, anchor="w")
 
    process_tree.pack(fill="both", expand=True)
 
    # 刷新進(jìn)程列表按鈕
    refresh_button = tk.Button(root, text="刷新進(jìn)程列表", command=refresh_processes, font=("Arial", 12), bg="lightgreen")
    refresh_button.pack(pady=10)
 
    # 結(jié)束進(jìn)程按鈕
    kill_button = tk.Button(root, text="結(jié)束進(jìn)程", command=kill_process, font=("Arial", 12), bg="lightcoral")
    kill_button.pack(pady=10)
 
    # 導(dǎo)出進(jìn)程列表按鈕
    export_button = tk.Button(root, text="導(dǎo)出進(jìn)程列表", command=export_processes, font=("Arial", 12), bg="lightyellow")
    export_button.pack(pady=10)
 
    # 初始化進(jìn)程列表和內(nèi)存使用信息
    refresh_processes()
 
    root.mainloop()
 
if __name__ == "__main__":
    run_as_admin()
    create_gui()