import os
import shutil
import wx
 
class FileOrganizer(wx.Frame):
    def __init__(self, parent, title):
        super(FileOrganizer, self).__init__(parent, title=title, size=(500, 300))
 
        panel = wx.Panel(self)
        self.current_dir = os.getcwd()
 
        # 創(chuàng)建按鈕用來(lái)選擇文件夾
        select_folder_btn = wx.Button(panel, label="選擇文件夾", pos=(10, 10))
        select_folder_btn.Bind(wx.EVT_BUTTON, self.on_select_folder)
 
        # 創(chuàng)建按鈕用來(lái)開(kāi)始整理文件夾
        organize_btn = wx.Button(panel, label="整理文件夾", pos=(10, 50))
        organize_btn.Bind(wx.EVT_BUTTON, self.on_organize)
 
        # 創(chuàng)建文本框顯示當(dāng)前文件夾路徑
        self.dir_text = wx.StaticText(panel, label=self.current_dir, pos=(10, 100))
 
        self.Show()
 
    def on_select_folder(self, event):
        dlg = wx.DirDialog(self, "選擇文件夾", style=wx.DD_DEFAULT_STYLE)
        if dlg.ShowModal() == wx.ID_OK:
            self.current_dir = dlg.GetPath()
            self.dir_text.SetLabel(self.current_dir)
        dlg.Destroy()
 
    def on_organize(self, event):
        # 創(chuàng)建文件夾
        photos_dir = os.path.join(self.current_dir, "photos")
        if not os.path.exists(photos_dir):
            os.makedirs(photos_dir)
 
        documents_dir = os.path.join(self.current_dir, "documents")
        if not os.path.exists(documents_dir):
            os.makedirs(documents_dir)
 
        videos_dir = os.path.join(self.current_dir, "videos")
        if not os.path.exists(videos_dir):
            os.makedirs(videos_dir)
 
        shortcuts_dir = os.path.join(self.current_dir, "shortcuts")
        if not os.path.exists(shortcuts_dir):
            os.makedirs(shortcuts_dir)
 
        # 遍歷文件夾
        for filename in os.listdir(self.current_dir):
            filepath = os.path.join(self.current_dir, filename)
            if os.path.isfile(filepath):
                ext = os.path.splitext(filename)[1].lower()
                if ext in (".jpg", ".jpeg", ".png", ".gif"):
                    shutil.move(filepath, os.path.join(photos_dir, filename))
                elif ext in (".doc", ".docx", ".pdf", ".txt"):
                    shutil.move(filepath, os.path.join(documents_dir, filename))
                elif ext in (".mp4", ".avi", ".mov", ".wmv"):
                    shutil.move(filepath, os.path.join(videos_dir, filename))
                elif ext == ".lnk":
                    shutil.move(filepath, os.path.join(shortcuts_dir, filename))
 
        wx.MessageBox("文件夾整理完成！", "提示", wx.OK | wx.ICON_INFORMATION)
 
if __name__ == "__main__":
    app = wx.App()
    FileOrganizer(None, title="文件整理工具")
    app.MainLoop()

源代碼說(shuō)明

在該代碼中，我們創(chuàng)建了一個(gè)wxPython的GUI界面，包含了兩個(gè)按鈕和一個(gè)文本框。點(diǎn)擊“選擇文件夾”按鈕可以彈出一個(gè)對(duì)話框用來(lái)選擇需要整理的文件夾，點(diǎn)擊“整理文件夾”按鈕可以開(kāi)始整理文件夾。

首先，我們創(chuàng)建了四個(gè)文件夾：photos、documents、videos、shortcuts。如果這些文件夾不存在，我們就使用os.makedirs()函數(shù)創(chuàng)建這些文件夾。

然后，我們使用os.listdir()函數(shù)遍歷文件夾中的所有文件。如果文件是一個(gè)文件而不是文件夾，我們就獲取文件的擴(kuò)展名，并根據(jù)擴(kuò)展名將該文件移動(dòng)到相應(yīng)的文件夾中。我們使用shutil.move()函數(shù)將文件從原始位置移動(dòng)到新的位置。

最后，我們使用wx.MessageBox()函數(shù)在完成整理后彈出一個(gè)提示框。

請(qǐng)注意，該代碼只能處理一級(jí)目錄下的文件，如果需要處理子目錄中的文件，需要使用遞歸函數(shù)來(lái)實(shí)現(xiàn)。