基于Python實現(xiàn)一個文件夾整理工具

更新時間：2023年10月18日 09:48:15 作者：微小冷

這篇文章主要為大家詳細介紹了如何基于Python實現(xiàn)一個簡單的文件夾整理工具,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下

簡介

我們的目的是做一個像下面這樣的工具，前面兩個輸入框，用于輸入源路徑和目標路徑，下面的圖片、視頻、音樂表示在目標路徑中創(chuàng)建的文件夾，后面的文件后綴，表示將這類文件移動到對應的文件夾中，加減號可以新增或刪除文件夾。

文件夾對話框

第一步，實現(xiàn)源路徑和目標路徑的輸入輸出，從而需要自定義一個組件，代碼如下，其及具體實現(xiàn)邏輯，可以參考：自定義文件選擇按鈕

import tkinter as tk
import tkinter.ttk as ttk
from tkinter.filedialog import (askopenfilename,
    askopenfilenames, askdirectory, asksaveasfilename)
from tkinter.colorchooser import askcolor

class DialogButton(ttk.Frame):
    def __init__(self, master, 
        height, widthL, widthR, logtype, label=None, text=None, 
        frmDct={}, btnDct={}, enyDct={}, logDct={}):
        w = widthL + widthR
        super().__init__(master, 
            height=height, width = w, **frmDct)
        self.pack(fill=tk.X)

        self.text = tk.StringVar() if not text else text
        ttk.Entry(self, width=widthL, textvariable=self.text, 
            **enyDct).pack(side=tk.LEFT, fill = tk.X, expand=True)
        
        ttk.Button(self, width=widthR, 
            text=self.setLabel(logtype, label),
            command = self.Click, **btnDct).pack(side=tk.RIGHT, padx=5)
        self.logtype = logtype
        self.logDct = logDct

    def setLabel(self, key, label=None):
        if label:
            return label
        labelDct = {
            "文件"   : "選擇文件",
            "文件夾" : "選擇路徑",
            "多文件" : "選擇多個文件",
            "保存" : "存儲路徑",
            "顏色"   : "選擇顏色",
        }
        return labelDct[key]

    def Click(self):
        typeDct = {
            "文件"  : askopenfilename,
            "文件夾": askdirectory,
            "多文件": askopenfilenames,
            "保存"  : asksaveasfilename,
            "顏色"  : askcolor,
        }
        text = typeDct[self.logtype](**self.logDct)
        if self.logtype == "顏色":
            text = text[1]
        self.text.set(text)

    def get(self):
        return self.text.get()

    def set(self, txt):
        self.text.set(txt)

文件映射組件

關(guān)于文件夾整理這個功能，此前其實做過無圖形界面的版本：

Python整理下載文件夾，第一步自然是做一個文件映射字典，

F_DCT = {
    "圖片"  : [".jpg", ".png", ".jpeg", ".gif"],
    "視頻"  : [".mp4", ".mkv"],
    "音樂"  : [".mp3", ".wav"],
    "文檔"  : [".pdf"],  
    "文獻"  : [".caj"],
    "文本"  : [".csv", ".txt"],
    "源碼"  : [".py", ".c", ".cpp", ".php"],
    "office": [".docx",".xlsx", ".pptx", ".doc", ".ppt", ".xls"],
    "壓縮包": [".zip", ".tgz", ".rar", ".tar", ".7z", ".gz"],
    "安裝包": [".msi", ".exe"],
}

然后就是工具的核心內(nèi)容，文件映射組件，主要由左右兩部分組成，左側(cè)是文件夾名，右側(cè)是對應的文件后綴。從這兩個子組件出發(fā)，可以為其設置初始化參數(shù)以及方法。先看源碼，再逐段解析

import tkinter as tk
import tkinter.ttk as ttk
from ctrls import DialogButton
import os
import shutil
pJoin = os.path.join
pExist = os.path.exists

from pathlib import Path

class ExFolder(ttk.Frame):
    def __init__(self, master, 
        folder=None, ex=None, dst=None, **options):
        super().__init__(master, **options)
        self.pack()

        self.folder = tk.StringVar()
        if folder : self.setFolder(folder)
        self.ex = tk.StringVar()
        if ex : self.setEx(ex)
        self.setDst(dst)

        self.initWidgets()

    def initWidgets(self):
        ttk.Entry(self, textvariable=self.folder, 
            width=10).pack(side=tk.LEFT, padx=2)    
        ttk.Entry(self, textvariable=self.ex
            ).pack(side=tk.LEFT, fill=tk.X, expand=True)

    def getFolder(self):
        return self.folder.get()

    def setFolder(self, f):
        self.folder.set(f)
    
    def getEx(self):
        exs = self.ex.get().split(",")
        for i in range(len(exs)):
            exs[i] = exs[i].strip()
        return exs
    
    def setEx(self, exs):
        if type(exs)==list:
            exs = ', '.join([ex.strip() for ex in exs])
        self.ex.set(exs)
    
    def mvFile(self, src):
        try:
            shutil.move(src, self.dst)
        except Exception as e:
            self.err = print(e)
    
    # 設置目標文件夾
    def setDst(self, dst):
        if not dst:
            self.dst = None
            return
        self.dst = pJoin(dst, self.getFolder())
        if not pExist(self.dst):
            os.makedirs(self.dst)
        self.errs = {}

首先，folder和ex分別是左右兩個組件對應的可變文本，為了便于更新和調(diào)用，類中設計了兩組、四對便攜方法getFolder, setFolder和getEx和setEx。由于Folder中的內(nèi)容就是文件夾本身，所以設置和讀取方法僅僅是StringVar的二次封裝，而ex中的內(nèi)容則不然，需要對文本進行分割，并且實現(xiàn)字符串到列表的轉(zhuǎn)換。

mvFile和setDst是組件的核心功能，分別用于設置目標路徑并移動。之所以為目標文件夾設置安排了一個獨立方法，是因為目標文件夾不一定存在，可能涉及到遞歸創(chuàng)建的過程。

完整組件

最后，是整個組件的布局，源碼如下，其中撤銷功能并未實現(xiàn)。

class FolderTools(ttk.Frame):
    def __init__(self, master, fDct=F_DCT, **options):
        super().__init__(master, **options)
        self.pack()
        self.fDct = fDct

        self.initWidgets()
        self.fEnys = []
        self.exEnys = []

    def setDct(self, fDct):
        self.fDct = fDct
        self.exDct = {}
        for k,v in self.fDct.items():
            exDct.update({ex:k for ex in v})

    def initWidgets(self):
        self.exForders = []
        pDct = dict(side=tk.TOP, expand=True, fill=tk.X)
        self.srcFolder = DialogButton(self, 5, 25, 8, 
                                      "文件夾", label="源路徑")
        self.srcFolder.pack(**pDct)
        self.dstFolder = DialogButton(self, 5, 25, 8, 
                                      "文件夾", label="目標路徑")
        self.dstFolder.pack(**pDct)

        btns = ttk.Frame(self)
        btns.pack(**pDct)
        ttk.Button(btns, text="? ", 
            command=self.btnAddFolder).grid(row=0,column=0)
        ttk.Button(btns, text="?", 
            command=self.btnDelFolder).grid(row=0,column=1)
        ttk.Button(btns, text="移動", 
            command=self.btnMove).grid(row=0,column=2)
        ttk.Button(btns, text="撤銷", 
            command=self.btnMove).grid(row=0,column=3)

        self.exFrm = ttk.LabelFrame(self, text="文件夾映射表")
        self.exFrm.pack(**pDct)

        for k, v in self.fDct.items():
            self.addFolders(self.exFrm, k, v)

    def btnAddFolder(self):
        self.addFolders(self.exFrm, None, None)

    def addFolders(self, frm, folder, exs):
        ef = ExFolder(frm, folder, exs)
        ef.pack(side=tk.TOP, expand=True, fill=tk.X, padx=2, pady=1)
        self.exForders.append(ef)

    def btnDelFolder(self):
        if len(self.exForders) == 0: return
        self.exForders[-1].pack_forget()
        del self.exForders[-1]
    
    def setExDct(self):
        self.exDct = {}
        for ef in self.exForders:
            self.exDct.update({key:ef for key in ef.getEx()})

    def btnMove(self):
        src = self.srcFolder.get()
        dst = self.dstFolder.get()
        if dst=="": dst = src

        # 設置后綴映射字典 ex：ExFolder
        exDct = {}
        for ef in self.exForders:
            exDct.update({ex:ef for ex in ef.getEx()})
            ef.setDst(dst) # 設置目標文件夾

        files = os.listdir(src)
        for f in files:
            p = pJoin(src, f)
            ex = Path(p).suffix.lower()
            if ex in exDct:
                exDct[ex].mvFile(p)

三個已經(jīng)實現(xiàn)的功能按鈕，?用于新增一個文件夾映射組件，?則移除最后一個映射組件。移除組件時需要注意，不僅僅要把組件解綁，還要把組件對應的實例刪掉。

核心功能被綁定在移動按鈕上，首先讀取源路徑和目標路徑，如果目標路徑未輸入，就在源路徑下工作。

接下來，將文件映射組件重新拆解為映射字典，最后調(diào)用對應組件的mvFile方法，實現(xiàn)所有文件的移動。

到此這篇關(guān)于基于Python實現(xiàn)一個文件夾整理工具的文章就介紹到這了,更多相關(guān)python文件夾整理工具內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: