Python自動化整理文件,一鍵完成!

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

通過本例還可以學(xué)到:os、glob、shutil三個標(biāo)準庫的綜合運用，現(xiàn)在讓我們一起來捋一捋實現(xiàn)的邏輯。

首先是導(dǎo)入需要的庫

import?os
import?shutil
import?glob

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

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

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

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

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參數(shù)的設(shè)置確保遍歷。由于需要找出所有的文件而非文件夾，這里用os.path.isfile進行判斷。最后可以輸出文件的絕對路徑先看看代碼有沒出現(xiàn)錯誤，讓我們繼續(xù)往下寫

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'

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

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

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}')

為了避免移動文件夾而造成的異常，尤其是系統(tǒng)盤，因此這里用的是復(fù)制。按照需要也可以換成shutil.move最后我們可以加上分類文件夾和所有文件的計數(shù)并輸出。完整代碼如下，拿走就能用！

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}個文件夾中')