二十行Python代碼給你的桌面文件分分類

大家好，本文跟大家分享一個文件整理腳本的實現過程。具體的功能很簡單，給定一個打算整理的文件夾目錄，這個腳本可以將該目錄下的所有文件都揪出來，並且根據後綴名歸類到不同的文件夾裏，就像下圖一樣，簡直是整理愛好者的福音

通過本例還可以學到:os、glob、shutil三個標準庫的綜合運用，現在讓我們一起來捋一捋實現的邏輯。

首先是導入需要的庫

import os
import shutil
import glob

os庫可以完成操作系統層面大量操作，例如文件夾的創建、移動、重命名、刪除等，有些功能實現的不是很完美，就用到shutil庫跟其互補了，例如文件的複製、移動等。glob庫可以利用通配符進行文件的搜索獲取，非常強大

# 設置建立分類總文件夾的路徑，這裏按自己的實際路徑修改
mkdir_path = r'C:\Users\chenx\文件夾分類'
# 設置需要遍歷整理的文件夾路徑，可以依據自己的實際需求修改
goal_dir = r'C:\xxxxxxxx)'

if not os.path.exists(mkdir_path):
    os.mkdir(mkdir_path)

os.mkdir可以在指定路徑創建文件夾，但如果文件夾已經存在則會報錯，因此謹慎一點可以利用os.path.exists先對文件夾的存在與否進行判斷，接下來是代碼核心循環，爲了方便理解先簡化成如下形式：

for file in glob.glob(f'{goal_dir}/**/*', recursive=True):
    if os.path.isfile(file):
        print(file)

glob.glob(f'{goal_dir}/**/*', recursive=True)中**/*是通配符的重要用法，*可以代表任意個字符，包括0個字符，recursive參數的設置確保遍歷。由於需要找出所有的文件而非文件夾，這裏用os.path.isfile進行判斷。最後可以輸出文件的絕對路徑先看看代碼有沒出現錯誤，讓我們繼續往下寫

for file in glob.glob(f'{goal_dir}/**/*', recursive=True):
    if os.path.isfile(file):
        filename = os.path.basename(file)
        if '.' in filename:
            suffix = filename.split('.')[-1]
        else:
            suffix = 'others'

這裏發生了什麼呢？確認遍歷到的是文件後，先用os.path.basename獲取絕對路徑中的文件名，接下來就是獲取後綴名了。可以簡單用split根據.將字符串“劈開”，然後取最後一個元素就是後綴名了，但注意這裏必須要考慮一個特殊情況：有些文件沒有後綴名(文件類型就叫 文件)，且名字中也沒有.，這時用字符串方法split就會報錯。如下圖：

因此需要先判斷文件中有沒有.。由於我們是利用後綴名建立文件夾，所以索性將文件名中沒有.的 文件 類型統一分類到others文件夾了，(這個實現邏輯大致上沒有問題，但是依然忽略了一種極端情況：有些文件沒有後綴名，且文件名中有個.，哈哈哈哈這種就會被上面的實現邏輯拆解出錯誤的後綴名了。更好的方法是有個函數可以直接獲取文件的後綴名，利用這個對文件進行分類，感興趣的讀者可以自己嘗試)，接下來就可以根據後綴名產生文件夾了，這裏依然要注意先判斷文件夾是否已經產生

for file in glob.glob(f'{goal_dir}/**/*', recursive=True):
    if os.path.isfile(file):
        filename = os.path.basename(file)
        if '.' in filename:
            suffix = filename.split('.')[-1]
        else:
            suffix = 'others'
        if not os.path.exists(f'{mkdir_path}/{suffix}'):
            os.mkdir(f'{mkdir_path}/{suffix}')
        shutil.copy(file, f'{mkdir_path}/{suffix}')

爲了避免移動文件夾而造成的異常，尤其是系統盤，因此這裏用的是複製。按照需要也可以換成shutil.move最後我們可以加上分類文件夾和所有文件的計數並輸出。完整代碼如下，拿走就能用！

import os
import shutil
import glob

mkdir_path = r'C:\Users\chenx\文件夾分類'
goal_dir = r'C:\xxxxxxxx'

if not os.path.exists(mkdir_path):
    os.mkdir(mkdir_path)

file_num = 0
dir_num = 0

for file in glob.glob(f'{goal_dir}/**/*', recursive=True):
    if os.path.isfile(file):
        filename = os.path.basename(file)
        if '.' in filename:
            suffix = filename.split('.')[-1]
        else:
            suffix = 'others'
        if not os.path.exists(f'{mkdir_path}/{suffix}'):
            os.mkdir(f'{mkdir_path}/{suffix}')
            dir_num += 1
        shutil.copy(file, f'{mkdir_path}/{suffix}')
        file_num += 1

print(f'整理完成，有{file_num}個文件分類到了{dir_num}個文件夾中')