Python根據文件后綴實現文件夾整理

更新時間：2024年02月06日 09:39:50 作者：微小冷

這篇文章主要為大家詳細介紹了Python如何根據文件后綴實現文件夾整理,文中的示例代碼講解詳細,具有一定的借鑒價值,有需要的可以參考下

文件類型字典

根據熵增原理，我們的文件夾在經過一段時間之后會變得混亂不堪，尤其是桌面和下載文件夾，找個東西會十分費勁，特別影響效率，所以用Python做一個文件夾整理小工具還是很有必要的。

而最常見的整理方法，就是按照文件類型，比如按照下表對文件進行歸類

文件夾	文件類別
圖片	“.jpg”, “.png”, “.jpeg”, “.gif”
視頻	“.mp4”, “.mkv”
音樂	“.mp3”, “.wav”
文檔	“.pdf”
文本	“.csv”, “.txt”
源碼	“.py”, “.c”, “.cpp”, “.php”
office	“.docx”,“.xlsx”, “.pptx”, “.doc”, “.ppt”, “.xls”
壓縮包	“.zip”, “.tgz”, “.rar”, “.tar”, “.7z”
安裝包	“.msi”, “.exe”

為了實現這個目標，可以先做一個文件類型字典，來綁定文件類型和后綴

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

而文件整理的邏輯是，先創(chuàng)建圖片、視頻等文件夾，然后通過文件的后綴找到其對應的文件類別，然后移動到相應的文件夾。為了更加便捷地實現這個功能，做一個反向的字典，用以通過后綴返回文件夾

exDct = {}
for key in fDct:
    exDct.update({ex:key for ex in fDct[key]})

移動文件

接下來就是移動文件，這里面有兩個問題，一是目標文件夾不存在，則需要新建；二則是目標文件夾中已經有了同名文件，那么則需要報錯

import os
import shutil
pJoin = os.path.join
pExist = os.path.exists

def mvFile(src, dst):
    if not pExist(dst):
        os.makedirs(dst)
    try:
        shutil.move(src, dst)
        return True
    except Exception as e:
        print(e)
        return False

主流程

接下來就是整個文件夾整理的主流程，第一步通過os.listdir得到所有文件名；第二步，遍歷所有文件名，如果這個文件名的后綴剛好就在我們的后綴字典中，那么就將這個文件移動到對應文件類型的文件夾里。

from pathlib import Path

# tar是準備整理的文件夾
def sortFiles(tar):
    files = os.listdir(tar)
    fsErr = []
    for f in files:
        src = pJoin(tar, f)
        ex = Path(src).suffix.lower()
        if ex not in exDct: continue
        dst = pJoin(tar, exDct[ex])
        if not mvFile(src, dst):
            fsErr.append(src)
    if len(fsErr) == 0:
        return "全部文件移動成功"
    else:
        return "以下文件移動失?。篭r\n"  + "\r\n".join(fsErr)

執(zhí)行效果如下，即下載文件夾中的文件都被移動到這幾個文件夾夾中了。