Python高級文件操作之shutil庫詳解

前言

什么算是高層的文件操作呢？

普通的文件操作，我們一般只涉及創(chuàng)建文件，文件夾以及寫入文件等等。假如我現(xiàn)在需要復制一個文件的內(nèi)容到另一個文件之中，用pathlib等都只能先打開復制文件，然后進行將其讀出來保存，然后再寫入新的文件，這種普通的復制操作，無形之中增加了許多步驟。

而shutil庫可以直接完成復制符間的操作，同時還支持歸檔。本篇，將詳細介紹文件的高層次操作。

一、copyfile()

copyfile()函數(shù)用于將一個文件的內(nèi)容復制到另一個文件之中，準備的來說，它不是copy內(nèi)容，而是直接copy文件，并重命名。

示例如下：

import shutil

shutil.copyfile('week.csv', 'week_copy.csv')

很簡單，就一行代碼，第1個參數(shù)是需要copy的源文件，第2個參數(shù)是需要復制的新文件。運行之后，效果如下：

二、copy()與copy2()

不過，copyfile()函數(shù)用于復制文件有一個缺陷，該函數(shù)的原理是通過打開源文件進行讀取。所以，如果是某些特殊的文件，比如是UNIX設(shè)備節(jié)點，沒有權(quán)限會報IOERROR錯誤。

還需要注意的是，通過copyfile()函數(shù)復制的文件是一個新文件，它不包括舊文件的訪問時間與修改時間。

而copy2()函數(shù)可以復制文件后保留訪問時間，修改時間。示例如下：

import shutil
import os
import time

shutil.copy2('week.csv', 'week_copy.csv')
stat_info1 = os.stat('week.csv')
stat_info2 = os.stat('week_copy.csv')
print("源文件信息")
print(oct(stat_info1.st_mode))
print('文件創(chuàng)建時間:{}'.format(time.ctime(stat_info1.st_ctime)))
print('文件訪問時間:{}'.format(time.ctime(stat_info1.st_atime)))
print('文件修改時間:{}'.format(time.ctime(stat_info1.st_mtime)))
print("復制文件信息")
print(oct(stat_info2.st_mode))
print('文件創(chuàng)建時間:{}'.format(time.ctime(stat_info2.st_ctime)))
print('文件訪問時間:{}'.format(time.ctime(stat_info2.st_atime)))
print('文件修改時間:{}'.format(time.ctime(stat_info2.st_mtime)))

運行之后，效果如下：

而copy()與copyfile()一樣都是復制文件的函數(shù)，不過copy()你給它一個文件夾名或者文件都可以復制，如果給文件夾名稱，它會復制到文件夾下的目錄中，當然復制的文件同名。而copyfile()如果給的是一個文件夾名稱會報錯。

示例如下：

import shutil
import os

os.mkdir('data')
shutil.copy('week.csv', 'data')

運行之后，效果如下：

三、copymode()與copystat()

通過copy2()函數(shù)，我們知道了如何復制文件內(nèi)容和文件信息等操作。但是其復制并不包含文件的權(quán)限，比如我們的Win10系統(tǒng)C盤中，有些文件只能讀不能寫，那么如果將權(quán)限也復制過去呢？

答案是：copymode()函數(shù)，示例如下：

import shutil
import os, stat

with open("temp.txt", 'wt') as file:
    file.write("11111111111111111")
os.chmod("temp.txt", stat.S_IREAD)
shutil.copymode('temp_copymode.txt', 'temp.txt')
print(oct(os.stat('temp.txt').st_mode))
print(oct(os.stat('temp_copymode.txt').st_mode))

需要注意的是，copymode()復制的只是文件權(quán)限，不是文件，也就是被賦予權(quán)限的文件名必須存在后才能將某個文件的權(quán)限復制給它。比如這里temp_copymode.txt獲取temp.txt權(quán)限，那么temp_copymode.txt必須存在后才能操作。而且這只是復制權(quán)限，并沒有復制內(nèi)容。（stat.S_IXGRP組用戶組執(zhí)行權(quán)限）

至于copystat()函數(shù)，不僅復制權(quán)限，而且也復制了文件的信息。具體代碼如下：

import shutil
import os, stat
import time

with open("temp.txt", 'wt') as file:
    file.write("11111111111111111")
os.chmod("temp.txt", stat.S_IXGRP)
shutil.copystat('temp_copymode.txt', 'temp.txt')
stat_info1 = os.stat('temp.txt')
stat_info2 = os.stat('temp_copymode.txt')
print("源文件信息")
print(oct(stat_info1.st_mode))
print('文件創(chuàng)建時間:{}'.format(time.ctime(stat_info1.st_ctime)))
print('文件訪問時間:{}'.format(time.ctime(stat_info1.st_atime)))
print('文件修改時間:{}'.format(time.ctime(stat_info1.st_mtime)))
print("復制文件信息")
print(oct(stat_info2.st_mode))
print('文件創(chuàng)建時間:{}'.format(time.ctime(stat_info2.st_ctime)))
print('文件訪問時間:{}'.format(time.ctime(stat_info2.st_atime)))
print('文件修改時間:{}'.format(time.ctime(stat_info2.st_mtime)))

運行之后，效果如下：

四、復制整個文件夾

上面的所有復制都是針對單個文件的操作，但其實我們還會在實際的項目中，將整個文件夾復制到另一個位置。而且復制整個文件夾肯定會用到遞歸，不過shutil庫有一個更簡單的函數(shù)用于實現(xiàn)該操作：copytree()。

示例如下：

import shutil

shutil.copytree('./text', './text_copy')

這里一行代碼就可以完成整個文件夾的實現(xiàn)。需要注意的是，第2個參數(shù)為需要復制到的目錄，但該文件夾不能存在，copytree()函數(shù)會自動創(chuàng)建的，如果存在會報錯。

參數(shù)copy_function

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

示例如下：

import shutil

def verbose_copy(src, dst):
    if src.endswith(".py"):
        return shutil.copy2(src, dst)

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

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

參數(shù)ignore

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

import shutil

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

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

五、刪除整個文件夾

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

import shutil

shutil.rmtree('./text_copy')

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

六、移動文件夾或文件

除了復制文件與文件夾之外，我們在實際的項目中，還會移動整個某個文件或某個目錄到另一個位置，shutil庫提供了move()函數(shù)用于移動文件或文件夾。

示例如下：

import shutil

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

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

七、查找文件

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

示例如下：

import shutil

filename = shutil.which('python')
print(filename)

運行之后，效果如下：

八、歸檔

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

import shutil

for format,desc in shutil.get_archive_formats():
    print('{:5}:{}'.format(format,desc))

運行之后，效果如下：

簡單的理解就是支持的壓縮格式。

8.1 壓縮文件

我們先來看一個簡單的例子，將某個文件夾進行壓縮，代碼如下：

import shutil

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

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

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

8.2 解壓縮

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

import shutil

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

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

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

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

示例如下：

import shutil

(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))

運行之后，效果如下：

這里博主是獲取電腦E盤的信息，比如這里獲取到E盤總大小total為238G，已使用Used為43G，未使用Free：194G。一個健壯的程序必然很嚴謹，所以在操作文件時，特別是比較大的文件時，一定要判斷磁盤的大小之后再操作。

到此這篇關(guān)于Python高級文件操作之shutil庫詳解的文章就介紹到這了,更多相關(guān)Python shutil庫內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！