Python高級(jí)文件操作之shutil庫詳解

第Python高級(jí)文件操作之shutil庫詳解這里一行代碼就可以完成整個(gè)文件夾的實(shí)現(xiàn)。需要注意的是，第2個(gè)參數(shù)為需要復(fù)制到的目錄，但該文件夾不能存在，copytree()函數(shù)會(huì)自動(dòng)創(chuàng)建的，如果存在會(huì)報(bào)錯(cuò)。

參數(shù)copy_function

有時(shí)候，我們復(fù)制文件夾并不是需要將整個(gè)文件夾都復(fù)制過來，而是只需要指定的符合規(guī)則的文件，比如我們只需要某個(gè)文件夾下的所有后綴為py的文件，那么怎么篩選呢？

示例如下：

importshutil

defverbose_copy(src,dst):

ifsrc.endswith(".py"):

returnshutil.copy2(src,dst)

shutil.copytree('./text','./text_copy',copy_function=verbose_copy)

這里，我們通過copytree()函數(shù)的參數(shù)copy_function進(jìn)行參數(shù)，該參數(shù)提供一個(gè)函數(shù)用于篩選符合規(guī)則的文件進(jìn)行復(fù)制，比如這里，我們提供了一個(gè)函數(shù)，并判斷后綴是否是.py然后再創(chuàng)建，當(dāng)然，它會(huì)保存目錄結(jié)構(gòu)，但只復(fù)制.py后綴的文件。

參數(shù)ignore

copytree()不僅只有copy_function參數(shù)，還有一個(gè)ignore參數(shù)，它可以過濾掉某些文件，比如還是如上面一樣，但是我們不需要.py后綴的文件，其他文件都復(fù)制，那么可以這樣實(shí)現(xiàn)。

importshutil

shutil.copytree('./text','./text_copy',ignore=shutil.ignore_patterns('*.py'))

這樣，就不會(huì)復(fù)制*.py規(guī)則的文件。

五、刪除整個(gè)文件夾

既然有復(fù)制整個(gè)文件夾，那么肯定shutil庫也會(huì)提供反向的操作用于刪除整個(gè)文件夾。刪除整個(gè)文件夾的函數(shù)為：rmtree()。

importshutil

shutil.rmtree('./text_copy')

這樣，我們就刪除了上面復(fù)制的文件夾內(nèi)容。

六、移動(dòng)文件夾或文件

除了復(fù)制文件與文件夾之外，我們?cè)趯?shí)際的項(xiàng)目中，還會(huì)移動(dòng)整個(gè)某個(gè)文件或某個(gè)目錄到另一個(gè)位置，shutil庫提供了move()函數(shù)用于移動(dòng)文件或文件夾。

示例如下：

importshutil

shutil.move('week.csv','./text')

這里，我們將csv文件移動(dòng)到了剛才用于復(fù)制的text源文件目錄。（移動(dòng)文件夾一樣操作，只是將文件名改為文件夾名）

七、查找文件

在更多的文件操作中，我們往往還需要查找某個(gè)文件。而shutil庫提供了which()函數(shù)用于搜索查找目標(biāo)文件。它有3個(gè)參數(shù)：mode可以設(shè)置查找文件的權(quán)限，path為需要查找的路徑，cmd為要查找的文件。

示例如下：

importshutil

filename=shutil.which('python')

print(filename)

運(yùn)行之后，效果如下：

八、歸檔

shutil庫提供了更多高層函數(shù)來創(chuàng)建和解壓歸檔文件。我們先來看看shutil庫支持哪些格式，示例如下：

importshutil

forformat,descinshutil.get_archive_formats():

print('{:5}:{}'.format(format,desc))

運(yùn)行之后，效果如下：

簡(jiǎn)單的理解就是支持的壓縮格式。

8.1壓縮文件

我們先來看一個(gè)簡(jiǎn)單的例子，將某個(gè)文件夾進(jìn)行壓縮，代碼如下：

importshutil

shutil.make_archive('text','gztar',root_dir='..',base_dir='base_demo')

這里我們將text文件夾，壓縮格式為gztar。運(yùn)行之后，效果如下：

其中，root_dir指定要壓縮的路徑根目錄（默認(rèn)當(dāng)前目錄），只能指定路徑，優(yōu)先級(jí)低于base_dir。base_dir指定要壓縮文件的路徑，可以指定路徑下的文件名，也可以指定路徑。

8.2解壓縮

既然有壓縮，肯定也就有解壓縮。shutil庫提供了unpack_archive()進(jìn)行解壓縮文件。示例如下：

importshutil

shutil.unpack_archive('text.tar.gz',extract_dir='text_un')

unpack_archive()函數(shù)的第1個(gè)參數(shù)為需要解壓的文件，第2個(gè)參數(shù)為需要解壓到哪里。

8.3文件系統(tǒng)空間

完成一個(gè)長(zhǎng)時(shí)間運(yùn)行的可能耗盡可用空間的操作之前，最好先檢查本地文件系統(tǒng)，來看看有多少可用的空間。shutil庫提供了disk_usage()函數(shù)來返回包括總空間，當(dāng)前正在使用的空間以及未使用的空間。

示例如下：

importshutil

(total,used,free)=shutil.disk_usage("E:/")

BytesPerGB=1024*1024*1024

print("Total:%.2fGB"%(float(total)/BytesPerGB))

print("Used:%.2fGB"%(float(used)/BytesPerGB))

print("Free:%.2fGB"%(float(free)/BytesPerGB))

運(yùn)行之后，效