轉載請注明：陳熹 chenx6542@foxmail.com （簡書號：半為花間酒）
若公眾號內轉載請聯系公眾號：早起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}個文件夾中')