# 導(dǎo)入必要的庫
import os
import sys
import tkinter as tk
from tkinterdnd2 import TkinterDnD # 如果使用了 tkinterdnd2

def resource_path(relative_path):
    """
    獲取資源文件的絕對路徑，用于處理 PyInstaller 打包后的情況。
    這個函數(shù)會判斷當(dāng)前環(huán)境是開發(fā)環(huán)境還是 PyInstaller 打包后的環(huán)境，
    并返回正確的資源文件路徑。

    Args:
        relative_path (str): 資源文件相對于腳本或打包目錄的相對路徑。

    Returns:
        str: 資源文件的絕對路徑。
    """
    try:
        # PyInstaller 創(chuàng)建臨時文件夾，并將路徑存儲在 _MEIPASS 中
        # 如果程序正在 PyInstaller 創(chuàng)建的臨時環(huán)境中運行，sys._MEIPASS 會被設(shè)置
        base_path = sys._MEIPASS
    except Exception:
        # 如果不在 PyInstaller 環(huán)境中（即開發(fā)環(huán)境），使用當(dāng)前腳本的目錄
        base_path = os.path.abspath(".")
    # 返回完整的資源文件路徑
    return os.path.join(base_path, relative_path)

# --- 示例：在您的 Tkinter 應(yīng)用類中使用 ---
class MyApplication:
    def __init__(self, root):
        self.root = root
        self.root.title("我的 Tkinter 應(yīng)用")

        # ... 其他 GUI 初始化代碼 ...

        # 初始化完成后，設(shè)置窗口圖標(biāo)
        self.set_window_icon()

    def set_window_icon(self):
        """
        設(shè)置 Tkinter 窗口的圖標(biāo)。
        """
        # 假設(shè)您的圖標(biāo)文件名為 'my_app_icon.ico'，并放在與腳本同級目錄
        icon_filename = "my_app_icon.ico" # 也可以是子目錄路徑，如 "assets/my_app_icon.ico"
        icon_path = resource_path(icon_filename)

        try:
            # 檢查圖標(biāo)文件是否存在
            if os.path.exists(icon_path):
                # 使用構(gòu)建的完整路徑設(shè)置圖標(biāo)
                self.root.iconbitmap(icon_path)
                print(f"窗口圖標(biāo)設(shè)置成功: {icon_path}")
            else:
                print(f"警告：圖標(biāo)文件未找到: {icon_path}")
                # 可以選擇不設(shè)置圖標(biāo)，或者顯示一個錯誤提示
        except tk.TclError as e:
            print(f"設(shè)置窗口圖標(biāo)失敗 (TclError): {e}")
            # 圖標(biāo)格式錯誤或不支持時會拋出 TclError
        except Exception as e:
            print(f"設(shè)置窗口圖標(biāo)失敗 (其他錯誤): {e}")

# --- 主程序入口 ---
if __name__ == "__main__":
    # 如果使用了 tkinterdnd2，必須使用 TkinterDnD.Tk()
    root = TkinterDnD.Tk()
    # 創(chuàng)建應(yīng)用實例
    app = MyApplication(root)
    # 啟動 GUI 主循環(huán)
    root.mainloop()

# Windows 示例
pyinstaller --onefile --windowed --icon=my_app_icon.ico --add-data "my_app_icon.ico;." your_script.py

# Linux/macOS 示例
pyinstaller --onefile --windowed --icon=my_app_icon.ico --add-data "my_app_icon.ico:." your_script.py