import wx
import os
 
class MainFrame(wx.Frame):
    def __init__(self, parent, title):
        wx.Frame.__init__(self, parent, title=title, size=(800, 600))
        
        # 創(chuàng)建一個(gè)垂直布局的主容器，并將其設(shè)置為窗體的主布局管理器
        main_sizer = wx.BoxSizer(wx.VERTICAL)
        self.SetSizer(main_sizer)
        
        # 創(chuàng)建一個(gè)水平布局的容器，用于將樹(shù)狀組件和預(yù)覽窗格放置在同一行
        hbox = wx.BoxSizer(wx.HORIZONTAL)
        main_sizer.Add(hbox, proportion=1, flag=wx.EXPAND|wx.ALL, border=5)
        
        # 創(chuàng)建一個(gè)按鈕，用于打開(kāi)文件夾選擇對(duì)話框
        button = wx.Button(self, wx.ID_ANY, "Open Folder")
        hbox.Add(button, proportion=0, flag=wx.ALL, border=5)
        
        # 創(chuàng)建一個(gè)樹(shù)狀組件，用于顯示指定目錄中的所有文件
        self.tree = wx.TreeCtrl(self, style=wx.TR_DEFAULT_STYLE|wx.TR_HIDE_ROOT)
        self.root = self.tree.AddRoot("Root")
        hbox.Add(self.tree, proportion=1, flag=wx.EXPAND|wx.ALL, border=5)
        
        # 創(chuàng)建一個(gè)預(yù)覽窗格，用于顯示選定的圖像文件
        self.preview = wx.StaticBitmap(self, wx.ID_ANY, wx.NullBitmap)
        hbox.Add(self.preview, proportion=1, flag=wx.EXPAND|wx.ALL, border=5)
        
        # 將按鈕的單擊事件綁定到處理函數(shù)self.on_open_folder()上
        button.Bind(wx.EVT_BUTTON, self.on_open_folder)
        
        # 將樹(shù)狀組件的選中事件綁定到處理函數(shù)self.on_tree_select()上
        self.Bind(wx.EVT_TREE_SEL_CHANGED, self.on_tree_select, self.tree)
        
        # 顯示窗體
        self.Centre()
        self.Show(True)
    
    def create_tree(self, parent, path):
        # 遞歸地向樹(shù)狀組件中添加指定目錄中的所有文件和子目錄
        for item in os.listdir(path):
            fullpath = os.path.join(path, item)
            if os.path.isdir(fullpath):
                node = self.tree.AppendItem(parent, item)
                self.create_tree(node, fullpath)
            else:
                ext = os.path.splitext(fullpath)[1].lower()
                if ext in [".jpg", ".jpeg", ".png", ".bmp", ".gif"]:
                    self.tree.AppendItem(parent, item, data=fullpath)
    
    def on_open_folder(self, event):
        # 處理打開(kāi)文件夾按鈕的單擊事件，打開(kāi)文件夾選擇對(duì)話框并更新樹(shù)狀組件
        dialog = wx.DirDialog(self, "Select a folder")
        if dialog.ShowModal() == wx.ID_OK:
            self.tree.DeleteChildren(self.root)
            self.create_tree(self.root, dialog.GetPath())
        dialog.Destroy()
                
    def on_tree_select(self, event):
        # 處理樹(shù)狀組件的選中事件，更新預(yù)覽窗格中的圖像文件
        item = event.GetItem()
        if item:
            filepath = self.tree.GetItemData(item)
            if filepath:
                image = wx.Image(filepath, wx.BITMAP_TYPE_ANY)
                self.preview.SetBitmap(wx.Bitmap(image))
                
if __name__ == "__main__":
    app = wx.App(False)
    frame = MainFrame(None, "File Viewer")
    app.MainLoop()