python自動(dòng)化腳本辦公-文件整理

今天講解文件整理腳本的實(shí)現(xiàn)過程。這是一個(gè)很有用的技能，可以幫助你管理你的電腦上的各種文件。需求如下：

• 需求內(nèi)容：給定一個(gè)打算整理的文件夾目錄，這個(gè)腳本可以將該目錄下的所有文件都揪出來，并且根據(jù)后綴名歸類到不同的文件夾里。
• python能力：使用python的內(nèi)置模塊os和shutil來操作文件和目錄，使用glob模塊來匹配文件名，使用with語句來打開和關(guān)閉文件。
• 實(shí)現(xiàn)步驟：
  • 導(dǎo)入需要的模塊
  • 定義要整理的目錄和要?jiǎng)?chuàng)建的子目錄
  • 遍歷目錄下的所有文件，獲取文件名和后綴名
  • 根據(jù)后綴名判斷文件屬于哪個(gè)子目錄，如果不存在則創(chuàng)建
  • 使用shutil模塊的move函數(shù)將文件移動(dòng)到對(duì)應(yīng)的子目錄
  • 打印完成信息

教程面對(duì)的是小白級(jí)別的讀者，所以我會(huì)盡量用簡單明了的語言來解釋每一行代碼的作用，并且給出完整的代碼示例。我希望你能跟著我一起學(xué)習(xí)，并且在自己的電腦上運(yùn)行和測試這個(gè)腳本。

導(dǎo)入python的內(nèi)置模塊

首先，我們需要導(dǎo)入一些python的內(nèi)置模塊，它們可以幫助我們操作文件和目錄。os模塊提供了一些基本的文件系統(tǒng)功能，比如獲取文件路徑，創(chuàng)建和刪除目錄等。shutil模塊提供了一些高級(jí)的文件操作功能，比如復(fù)制和移動(dòng)文件等。glob模塊可以讓我們用通配符來匹配文件名，比如*.txt可以匹配所有后綴名為txt的文件。with語句可以讓我們方便地打開和關(guān)閉文件，不用擔(dān)心忘記關(guān)閉或者出現(xiàn)異常。下面是導(dǎo)入模塊的代碼：

# 導(dǎo)入需要的模塊
import os
import shutil
import glob

定義要整理的目錄和要?jiǎng)?chuàng)建的子目錄

接下來，我們需要定義要整理的目錄和要?jiǎng)?chuàng)建的子目錄。我們假設(shè)要整理的目錄是D:test，里面有各種類型的文件，比如文本文件，圖片文件，音頻文件等。我們想要把這些文件按照后綴名歸類到不同的子目錄里，比如txt文件放到D:testtxt，jpg文件放到D:testjpg，mp3文件放到D:testmp3等。我們可以用一個(gè)字典來存儲(chǔ)這些子目錄的信息，鍵是后綴名，值是對(duì)應(yīng)的子目錄路徑。下面是定義目錄的代碼：

# 定義要整理的目錄和要?jiǎng)?chuàng)建的子目錄
source_dir = "D:test" # 要整理的目錄
sub_dirs = { # 子目錄信息
    ".txt": "D:testtxt",
    ".jpg": "D:testjpg",
    ".mp3": "D:testmp3"
}

遍歷目錄下的所有文件，獲取文件名和后綴名

然后，我們需要遍歷要整理的目錄下的所有文件，獲取文件名和后綴名。我們可以用os模塊的listdir函數(shù)來獲取目錄下的所有文件和子目錄，然后用os模塊的path.splitext函數(shù)來分割文件名和后綴名。我們還需要判斷是否是文件，因?yàn)閘istdir函數(shù)會(huì)返回子目錄，我們不需要處理它們。我們可以用os模塊的path.isfile函數(shù)來判斷是否是文件。下面是遍歷目錄的代碼：

# 遍歷目錄下的所有文件，獲取文件名和后綴名
for item in os.listdir(source_dir): # 遍歷目錄
    if os.path.isfile(os.path.join(source_dir, item)): # 判斷是否是文件
        file_name, file_ext = os.path.splitext(item) # 分割文件名和后綴名
        print(file_name, file_ext) # 打印文件名和后綴名

根據(jù)后綴名判斷文件屬于哪個(gè)子目錄

接著，我們需要根據(jù)后綴名判斷文件屬于哪個(gè)子目錄，如果不存在則創(chuàng)建。我們可以用字典的get方法來根據(jù)鍵獲取值，如果鍵不存在則返回None。我們可以用os模塊的path.exists函數(shù)來判斷子目錄是否存在，如果不存在則用os模塊的makedirs函數(shù)來創(chuàng)建。下面是判斷和創(chuàng)建子目錄的代碼：

# 根據(jù)后綴名判斷文件屬于哪個(gè)子目錄，如果不存在則創(chuàng)建
        sub_dir = sub_dirs.get(file_ext) # 根據(jù)后綴名獲取子目錄路徑
        if sub_dir: # 如果子目錄路徑存在
            if not os.path.exists(sub_dir): # 如果子目錄不存在
                os.makedirs(sub_dir) # 創(chuàng)建子目錄
                print(f"Created {sub_dir}") # 打印創(chuàng)建信息

將文件移動(dòng)到對(duì)應(yīng)的子目錄

接下來，我們需要使用shutil模塊的move函數(shù)將文件移動(dòng)到對(duì)應(yīng)的子目錄。我們需要提供源文件路徑和目標(biāo)文件路徑作為參數(shù)。我們可以用os模塊的path.join函數(shù)來拼接路徑。下面是移動(dòng)文件的代碼：

# 使用shutil模塊的move函數(shù)將文件移動(dòng)到對(duì)應(yīng)的子目錄
            source_file = os.path.join(source_dir, item) # 源文件路徑
            target_file = os.path.join(sub_dir, item) # 目標(biāo)文件路徑
            shutil.move(source_file, target_file) # 移動(dòng)文件
            print(f"Moved {source_file} to {target_file}") # 打印移動(dòng)信息

打印完成信息

最后，我們需要打印完成信息，告訴用戶整理工作已經(jīng)完成。下面是打印完成信息的代碼：

# 打印完成信息
print("Done!")

這樣，我們就完成了一個(gè)簡單的文件整理腳本。完整的代碼如下：

# 導(dǎo)入需要的模塊
import os
import shutil
import glob

# 定義要整理的目錄和要?jiǎng)?chuàng)建的子目錄
source_dir = "D:	est" # 要整理的目錄
sub_dirs = { # 子目錄信息
    ".txt": "D:	est	xt",
    ".jpg": "D:	estjpg",
    ".mp3": "D:	estmp3"
}

# 遍歷目錄下的所有文件，獲取文件名和后綴名
for item in os.listdir(source_dir): # 遍歷目錄
    if os.path.isfile(os.path.join(source_dir, item)): # 判斷是否是文件
        file_name, file_ext = os.path.splitext(item) # 分割文件名和后綴名
        print(file_name, file_ext) # 打印文件名和后綴名

        # 根據(jù)后綴名判斷文件屬于哪個(gè)子目錄，如果不存在則創(chuàng)建
        sub_dir = sub_dirs.get(file_ext) # 根據(jù)后綴名獲取子目錄路徑
        if sub_dir: # 如果子目錄路徑存在
            if not os.path.exists(sub_dir): # 如果子目錄不存在
                os.makedirs(sub_dir) # 創(chuàng)建子目錄
                print(f"Created {sub_dir}") # 打印創(chuàng)建信息

            # 使用shutil模塊的move函數(shù)將文件移動(dòng)到對(duì)應(yīng)的子目錄
            source_file = os.path.join(source_dir, item) # 源文件路徑
            target_file = os.path.join(sub_dir, item) # 目標(biāo)文件路徑
            shutil.move(source_file, target_file) # 移動(dòng)文件
            print(f"Moved {source_file} to {target_file}") # 打印移動(dòng)信息

# 打印完成信息
print("Done!")

如果你在自己的電腦上運(yùn)行這個(gè)腳本，你會(huì)看到類似下面的輸出：

test1 .txt
test2 .jpg
test3 .mp3
Created D:testtxt
Moved D:testtest1.txt to D:testtxttest1.txt
Created D:testjpg
Moved D:testtest2.jpg to D:testjpgtest2.jpg
Created D:testmp3
Moved D:testtest3.mp3 to D:testmp3test3.mp3
Done!