[Python] 自動(dòng)化辦公 批量命名和移動(dòng)文件

轉(zhuǎn)載請注明:陳熹 chenx6542@foxmail.com (簡書號:半為花間酒)
若公眾號內(nèi)轉(zhuǎn)載請聯(lián)系公眾號:早起Python

這篇文章能學(xué)到的主要內(nèi)容:
基于 os globshutil 對文件管理的綜合運(yùn)用

一、需求描述

你的電腦中有一個(gè)文件夾 files1 存放著 1800+ 個(gè)文件,如下所示:

需要完成的內(nèi)容是:將 1835 個(gè)文件移動(dòng)到新文件夾 file2,并且重命名文件,名字開頭加上 序號 和 “終稿” 兩個(gè)字,如名字更改為 “1-終稿-xxxxx(原文件名)”

你心里可能想著:這是人做的事???但確實(shí)這是真實(shí)的需求,文件批量重命名非常常見,如果沒有一些技巧,那么只能耗費(fèi)大量的時(shí)間和人力去做。這里的技巧,就是 Python

另外還有一個(gè)問題:要先移動(dòng)再重命名還是先重命名再移動(dòng)呢?繼續(xù)往下看

二、前置知識和數(shù)據(jù)準(zhǔn)備

1. 生成大量隨機(jī)文件

真實(shí)的辦公場景并不會這樣的需求,畢竟誰想要無端給自己的電腦產(chǎn)生大量無用文件呢(也不要給別人的電腦亂用)
不得不提,生成隨機(jī)文件能夠幫助我們更好的測試自己 Python 文件管理的技能。如果你沒有合適的文件夾和文件夾供自己練習(xí),那么為什么不自己寫個(gè)代碼產(chǎn)生呢?
當(dāng)然,在這個(gè)過程中我們也會學(xué)習(xí)一些知識點(diǎn),先看代碼:

import random
import string

for i in range(2000):
    random_str = ''.join(random.sample(string.ascii_letters + string.digits, random.randint(1, 11)))
    file = open(r"C:\\xxx\\file1" + random_str + ".txt", 'w+') # 前面路徑是產(chǎn)生文件的目標(biāo)文件夾
    file.write(''.join(random.sample(string.ascii_letters + string.digits, random.randint(1, 11))))
    file.close()

通過 string 就可以獲得所有的字母和數(shù)字,利用 random.sample() 常規(guī)接受兩個(gè)參數(shù),一個(gè)是抽樣的范圍,一個(gè)是抽樣的次數(shù),默認(rèn)是放回抽樣。這樣就可以在給定的字母數(shù)字范圍內(nèi)隨機(jī)抽取 1-10 個(gè),但是返回的結(jié)果注意是列表,需要再用 .join 方法完成字符串拼接
用隨機(jī)產(chǎn)生的名字生成文件后,再在其內(nèi)部用類似的方法隨機(jī)寫入一些內(nèi)容:

上面的寫法不夠優(yōu)雅,因?yàn)樾枰涮资褂?file.close() 釋放,更好的方法是直接利用上下文管理器 with 結(jié)構(gòu),減少出錯(cuò)的幾率:

import random
import string

for i in range(2000):
    random_str = ''.join(random.sample(string.ascii_letters + string.digits, random.randint(1, 11)))
    with open(r"C:\\xxx\\file1" + random_str + ".txt", 'w+') as file:
        file.write(''.join(random.sample(string.ascii_letters + string.digits, random.randint(1, 11))))

因?yàn)榧词故请S機(jī)產(chǎn)生名字,但抽樣的范圍和次數(shù)不大決定了 2000 次抽樣會有一些抽簽組合成的名字完全一樣,后面形成的文件會覆蓋之前產(chǎn)生的文件,最終導(dǎo)致產(chǎn)生的文件沒有 2000 個(gè)

2. 重命名文件/文件夾

需要用到內(nèi)置庫 osos.rename() 方法

import os
os.rename('practice.txt', 'practice_rename.txt')  # 重命名文件
os.rename('文件夾1', '文件夾2') # 重命名文件夾

雖然需求中有重命名文件的需求,但實(shí)際上并不需要直接借助這個(gè)方法

3. 移動(dòng)文件/文件夾

需要用到內(nèi)置庫 shutilshutil.move 方法

import shutil 
shutil.move(r'.\practice.txt', r'.\文件夾1/') 
shutil.move(r'.\practice.txt', r'.\文件夾1/new.txt')

注意到上面后兩行代碼的區(qū)別嗎?前一行是將目標(biāo)文件移動(dòng)到目標(biāo)文件夾里,而后一行,在將目標(biāo)文件移動(dòng)到目標(biāo)文件夾里的同時(shí),能夠?qū)ζ溥M(jìn)行重命名
也就是說,我們并不需要用 os.rename 先命名文件再用 shutil.move 將其移動(dòng)的指定文件夾,而是可以用 shutil.move 一步到位

4. 遍歷獲取文件

采用基于 glob 庫的迭代框架:

import glob
path = xxx

for file in glob.glob(f'{path}/**/*.xlsx', recursive=True): 
    pass

上面的代碼能夠獲取給定路徑內(nèi)部所有文件夾下的 Excel 文件(.xlsx 格式), recursive 參數(shù)默認(rèn)為 False,當(dāng)為 True 時(shí)允許逐級遍歷

而本例需要獲取給定文件夾下的所有 .txt 文件,則更加簡單:

import glob
path = xxx

for file in glob.glob(f'{path}/*.txt'): 
    pass

3. 代碼實(shí)現(xiàn)

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

import os
import shutil
import glob

path = r"C:\xxx" # 存放大量需更名移動(dòng)文件的文件夾路徑的上一級路徑

上文提到,不需要利用 os.rename 那為什么要導(dǎo)入 os 庫呢?
一方面因?yàn)橐ㄟ^這個(gè)庫產(chǎn)生新的文件夾。也可以手動(dòng)完成,但交給代碼多了判斷也不容易出錯(cuò):

if not os.path.exists(path + r'\file2'): 
    os.mkdir(path + r'\file2')

另一方面下文還會用它獲取文件名
然后就可以移動(dòng)更名一步到位,glob 迭代文件框架遍歷獲取文件絕對路徑:

count = 1 # 生成序號
for file in glob.glob(f'{path}\\測試\\*.txt'):
    # 這里是文件絕對路徑,可以用字符串方法直接替換修改,但為了方便理解我還是用路徑拼接
    filename = os.path.basename(file)
    shutil.move(file, path + r'\file2' + f'\{count}-終稿-{filename}')
    count += 1

至此,需求成功完成!

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容