五一快到了,工作想劃水?十個(gè)Python辦公自動(dòng)化操作,拿走就用
一、OS模塊相關(guān)
1.遍歷文件夾
批量操作的前提就是對(duì)文件夾進(jìn)行遍歷,使用os模塊可以輕松的遍歷文件夾,os.walk 遍歷后產(chǎn)生三個(gè)參數(shù):
“當(dāng)前文件夾路徑包含文件夾名稱[列表形式]包含文件名稱[列表形式]”
代碼如下,大家可以根據(jù)自己的路徑進(jìn)行修改
- import os
- for dirpath, dirnames, filenames in os.walk(r'C:\\Program Files (x86)'):
- print(f'打開文件夾{dirpath}') # 當(dāng)前文件夾路徑
- if dirnames:
- print(dirnames) # 包含文件夾名稱[列表形式]
- if filenames:
- print(filenames) # 包含文件名稱[列表形式]
- print('-' * 10)
當(dāng)手上的需求明確有獲取給定路徑各級(jí)文件夾下所有符合條件的文件,并進(jìn)行相應(yīng)的批處理操作時(shí),即可使用 os.walk
2.目標(biāo)路徑是否是文件
有時(shí)我們需要判斷一個(gè)目錄下是否存在文件也可以使用os模塊。
給定一個(gè)目標(biāo)路徑 path ,通過一行代碼就能夠判斷這是文件還是文件夾路徑
- import os
- path = 'xxx'
- print(os.path.isfile(path))
3.獲取路徑中的文件名
os.path.basename 可以直接從絕對(duì)路徑中獲取最后的文件名,當(dāng)然如果用傳統(tǒng)的字符串切割方式也可以,即 path.split('\\')[-1]
- import os
- path = 'xxx'
- print(os.path.basename)
4.創(chuàng)建文件夾
創(chuàng)建文件夾的代碼非常常用,因?yàn)橥傻男挛募枷M袀€(gè)新的文件夾存儲(chǔ),代碼如下:
- import os
- dirpath = 'xxx'
- os.mkdir(dirpath)
但是,如果希望創(chuàng)建的文件夾已經(jīng)存在,再運(yùn)行 os.mkdir() 則會(huì)報(bào)錯(cuò)而終止代碼。為了避免這一情況的發(fā)生,可以在創(chuàng)建文件夾之前先判斷文件夾是否存在。
用到的代碼是 os.path.exists,只有當(dāng)路徑不存在(即 os.path.exists 返回的結(jié)果是 False時(shí)),才會(huì)創(chuàng)建:
- import os
- dirpath = 'xxx'
- if not os.path.exists(dirpath):
- os.mkdir(dirpath)
5.獲取桌面路徑
獲取桌面路徑也是非常常用的操作,可以使用os.path.join(os.path.expanduser("~"), 'Desktop')獲取桌面的絕對(duì)路徑。
這樣做的好處是可以把數(shù)據(jù)放在桌面上,在不同的電腦上都能調(diào)用代碼對(duì)數(shù)據(jù)進(jìn)行處理。如果是在一條電腦上把桌面路徑固定在字符串中,則換一臺(tái)電腦就必須修改桌面路徑。代碼如下:
- import os
- desktop_path = os.path.join(os.path.expanduser("~"), 'Desktop')
- print(desktop_path)
當(dāng)然把上面的代碼包裝成一個(gè)函數(shù) GetDesktopPath() 需要時(shí)調(diào)用它會(huì)更加方便
- import os
- def GetDesktopPath():
- return os.path.join(os.path.expanduser("~"), 'Desktop')
6.重命名文件/文件夾
需要用到 os.rename() 方法,下面的代碼示例中分別演示如何重命名文件和文件夾
- import os
- os.rename('practice.txt', 'practice_rename.txt') # 重命名文件
- os.rename('文件夾1', '文件夾2') # 重命名文件夾
7.批處理文件 - 1
除了前面的 os.walk 之外,有其他的 os 模塊下方法可完成獲取指定路徑的全部或符合條件的文件(非遍歷各級(jí)文件夾的需求),還可以使用下面兩個(gè)代碼第一種用到的方法是os.scandir(),使用如下:
- import os
- path = 'xxx'
- for file in os.scandir(path):
- print(file.name, file.path)
8.批處理文件 - 2
上面代碼最后輸出的是 給定路徑下各內(nèi)容的名字、絕對(duì)路徑第二種方法使用 os.listdir(),它比 os.scandir() 簡單一些,可直接調(diào)用輸出名稱而非路徑:
- import os
- path = 'xxx'
- for file in os.listdir(path):
- print(fil
二、shutil模塊相關(guān)
9.移動(dòng)文件/文件夾
shutil也是經(jīng)常出現(xiàn)在辦公自動(dòng)化場景中的模塊,我常用的就是移動(dòng)文件/文件夾。
需要用到shutil.move 方法,下面的代碼示例中分別演示如何移動(dòng)文件和文件夾:
- 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)行重命名
也就是說,如果我們需要移動(dòng)某個(gè)或某些文件到新的文件夾,并且需重命名文件,則我們并不需要用os.rename 先命名文件再用 shutil.move 將其移動(dòng)的指定文件夾,而是可以用 shutil.move 一步到位。
三、glob模塊相關(guān)
10.批處理文件 - 3
最后要介紹的是glob模塊,也是辦公自動(dòng)化必須要掌握的一個(gè)模塊,同樣可以用于批處理文件。
glob 最重要的功能就是搜索獲取同一級(jí)或者各子級(jí)下符合條件的文件(絕對(duì)路徑),非常適合寫批處理的代碼。
有時(shí)候我們需要對(duì)大量文件進(jìn)行相同操作,在寫完針對(duì)一份文件的操作后,只需要加上幾行代碼,就可以完成批處理全部文件的工作。大致代碼框架如下:
- import glob
- for file in glob.glob('**/*', recursive=True):
- print(file)
glob.glob() 是一個(gè)非常重要的方法,能夠獲取給定路徑下文件的絕對(duì)路徑,并且接受「通配符」搜索,大大拓寬了靈活程度,* 表示任意字符長度,**/* 的使用表示用通配符指代給定路徑下的任何一層,recursive 參數(shù)允許遍歷搜索。