【經驗分享】Python文件操作方法,看這篇就足夠

Python中有幾個內置模塊和方法來處理文件。這些方法被分割到例如os, os.path , shutilpathlib 等等幾個模塊中。文章將列舉Python中對文件最常用的操作和方法。

在這篇文章中,你將學習如何:

  • 獲取文件屬性
  • 創建目錄
  • 文件名模式匹配
  • 遍歷目錄樹
  • 創建臨時文件和目錄
  • 刪除文件和目錄
  • 複製、移動和重命名文件和目錄
  • 創建和解壓ZIP和TAR檔案
  • 使用fileinput 模塊打開多個文件

Python中文件數據的讀和寫

使用Python對文件進行讀和寫是十分簡單的。爲此,你首先必須使用合適的模式打開文件。這裏有一個如何打開文本文件並讀取其內容的例子。

with open('data.txt', 'r') as f:
    data = f.read()
    print('context: {}'.format(data))

open() 接收一個文件名和一個模式作爲它的參數,r 表示以只讀模式打開文件。想要往文件中寫數據的話,則用w 作爲參數。

在學習過程中有什麼不懂得可以加我的
python學習qun,855408893
羣裏有不錯的學習視頻教程、開發工具與電子書籍。  
與你分享python企業當下人才需求及怎麼從零基礎學習好python,和學習什麼內容

with open('data.txt', 'w') as f:
    data = 'some data to be written to the file'
    f.write(data)

在上述例子中,open()打開用於讀取或寫入的文件並返回文件句柄(本例子中的 f ),該句柄提供了可用於讀取或寫入文件數據的方法。


獲取目錄列表

假設你當前的工作目錄有一個叫 my_directory 的子目錄,該目錄包含如下內容:

.
├── file1.py
├── file2.csv
├── file3.txt
├── sub_dir
│   ├── bar.py
│   └── foo.py
├── sub_dir_b
│   └── file4.txt
└── sub_dir_c
    ├── config.py
    └── file5.txt

Python內置的 os 模塊有很多有用的方法能被用來列出目錄內容和過濾結果。爲了獲取文件系統中特定目錄的所有文件和文件夾列表,可以在遺留版本的Python中使用 os.listdir() 或 在Python 3.x 中使用 os.scandir() 。 如果你還想獲取文件和目錄屬性(如文件大小和修改日期),那麼 os.scandir() 則是首選的方法。

使用遺留版本的Python獲取目錄列表

import os
entries = os.listdir('my_directory')

os.listdir() 返回一個Python列表,其中包含path參數所指目錄的文件和子目錄的名稱。

['file1.py', 'file2.csv', 'file3.txt', 'sub_dir', 'sub_dir_b', 'sub_dir_c']

目錄列表現在看上去不容易閱讀,對 os.listdir() 的調用結果使用循環打印有助於查看。

for entry in entries:
    print(entry)

"""
file1.py
file2.csv
file3.txt
sub_dir
sub_dir_b
sub_dir_c
"""


使用現代版本的Python獲取目錄列表

在現代Python版本中,可以使用 os.scandir()pathlib.Path 來替代 os.listdir()

os.scandir() 在Python 3.5 中被引用,其文檔爲 PEP 471

os.scandir() 調用時返回一個迭代器而不是一個列表。

import os
entries = os.scandir('my_directory')
print(entries)
# <posix.ScandirIterator at 0x105b4d4b0>

ScandirIterator 指向了當前目錄中的所有條目。你可以遍歷迭代器的內容,並打印文件名。

import os
with os.scandir('my_directory') as entries:
    for entry in entries:
        print(entry.name)

這裏 os.scandir() 和with語句一起使用,因爲它支持上下文管理協議。使用上下文管理器關閉迭代器並在迭代器耗盡後自動釋放獲取的資源。在 my_directory 打印文件名的結果就和在 os.listdir() 例子中看到的一樣:

file1.py
file2.csv
file3.txt
sub_dir
sub_dir_b
sub_dir_c

另一個獲取目錄列表的方法是使用 pathlib 模塊:

from pathlib import Path

entries = Path('my_directory')
for entry in entries.iterdir():
    print(entry.name)

pathlib.Path() 返回的是 PosixPathWindowsPath 對象,這取決於操作系統。

pathlib.Path() 對象有一個 .iterdir() 的方法用於創建一個迭代器包含該目錄下所有文件和目錄。由 .iterdir() 生成的每個條目都包含文件或目錄的信息,例如其名稱和文件屬性。pathlib 在Python3.4時被第一次引入,並且是對Python一個很好的加強,它爲文件系統提供了面向對象的接口。

在上面的例子中,你調用 pathlib.Path() 並傳入了一個路徑參數。然後調用 .iterdir() 來獲取 my_directory 下的所有文件和目錄列表。

pathlib 提供了一組類,以簡單並且面向對象的方式提供了路徑上的大多數常見的操作。使用 pathlib 比起使用 os 中的函數更加有效。和 os 相比,使用 pathlib 的另一個好處是減少了操作文件系統路徑所導入包或模塊的數量。

運行上述代碼會得到如下結果:

file1.py
file2.csv
file3.txt
sub_dir
sub_dir_b
sub_dir_c

使用 pathlib.Path()os.scandir() 來替代 os.listdir() 是獲取目錄列表的首選方法,尤其是當你需要獲取文件類型和文件屬性信息的時候。pathlib.Path() 提供了在 osshutil 中大部分處理文件和路徑的功能,並且它的方法比這些模塊更加有效。我們將討論如何快速的獲取文件屬性。

函數 描述
os.listdir() 以列表的方式返回目錄中所有的文件和文件夾
os.scandir() 返回一個迭代器包含目錄中所有的對象,對象包含文件屬性信息
pathlib.Path().iterdir() 返回一個迭代器包含目錄中所有的對象,對象包含文件屬性信息

這些函數返回目錄中所有內容的列表,包括子目錄。這可能並總是你一直想要的結果,下一節將向你展示如何從目錄列表中過濾結果。

列出目錄中的所有文件

這節將向你展示如何使用 os.listdir()os.scandir()pathlib.Path() 打印出目錄中文件的名稱。爲了過濾目錄並僅列出 os.listdir() 生成的目錄列表的文件,要使用 os.path

import os

basepath = 'my_directory'
for entry in os.listdir(basepath):
    # 使用os.path.isfile判斷該路徑是否是文件類型
    if os.path.isfile(os.path.join(base_path, entry)):
        print(entry)

在這裏調用 os.listdir() 返回指定路徑中所有內容的列表,接着使用 os.path.isfile() 過濾列表讓其只顯示文件類型而非目錄類型。代碼執行結果如下:

file1.py
file2.csv
file3.txt

一個更簡單的方式來列出一個目錄中所有的文件是使用 os.scandir()pathlib.Path() :

import os

basepath = 'my_directory'
with os.scandir(basepath) as entries:
    for entry in entries:
        if entry.is_file():
            print(entry.name)

使用 os.scandir() 比起 os.listdir() 看上去更清楚和更容易理解。對 ScandirIterator 的每一項調用 entry.isfile() ,如果返回 True 則表示這一項是一個文件。上述代碼的輸出如下:

file1.py
file3.txt
file2.csv

接着,展示如何使用 pathlib.Path() 列出一個目錄中的文件:

from pathlib import Path

basepath = Path('my_directory')
for entry in basepath.iterdir():
    if entry.is_file():
        print(entry.name)

.iterdir() 產生的每一項調用 .is_file() 。產生的輸出結果和上面相同:

file1.py
file3.txt
file2.csv

如果將for循環和if語句組合成單個生成器表達式,則上述的代碼可以更加簡潔。

修改後的版本如下:

from pathlib import Path

basepath = Path('my_directory')
files_in_basepath = (entry for entry in basepath.iterdir() if entry.is_file())
for item in files_in_basepath:
    print(item.name)

上述代碼的執行結果和之前相同。本節展示使用 os.scandir()pathlib.Path() 過濾文件或目錄比使用 os.listdir()os.path 更直觀,代碼看起來更簡潔。

列出子目錄

如果要列出子目錄而不是文件,請使用下面的方法。現在展示如何使用 os.listdir()os.path() :

import os

basepath = 'my_directory'
for entry in os.listdir(basepath):
    if os.path.isdir(os.path.join(basepath, entry)):
        print(entry)

當你多次調用 os.path,join() 時,以這種方式操作文件系統就會變得很笨重。在我電腦上運行此代碼會產生以下輸出:

sub_dir
sub_dir_b
sub_dir_c

下面是如何使用 os.scandir()

import os

basepath = 'my_directory'
with os.scandir(basepath) as entries:
    for entry in entries:
        if entry.is_dir():
            print(entry.name)

與文件列表中的示例一樣,此處在 os.scandir() 返回的每一項上調用 .is_dir() 。如果這項是目錄,則 is_dir() 返回 True,並打印出目錄的名稱。輸出結果和上面相同:

sub_dir_c
sub_dir_b
sub_dir

下面是如何使用 pathlib.Path()

from pathlib import Path

basepath = Path('my_directory')
for entry in basepath.iterdir():
    if entry.is_dir():
        print(entry.name)

.iterdir() 迭代器返回的每一項上調用 is_dir() 檢查是文件還是目錄。如果該項是目錄,則打印其名稱,並且生成的輸出與上一示例中的輸出相同:

sub_dir_c
sub_dir_b
sub_dir


獲取文件屬性

Python可以很輕鬆的獲取文件大小和修改時間等文件屬性。可以通過使用 os.stat()os.scandir()pathlib.Path 來獲取。

os.scandir()pathlib.Path() 能直接獲取到包含文件屬性的目錄列表。這可能比使用 os.listdir() 列出文件然後獲取每個文件的文件屬性信息更加有效。

下面的例子顯示瞭如何獲取 my_directory 中文件的最後修改時間。以時間戳的方式輸出:

import os

with os.scandir('my_directory') as entries:
    for entry in entries:
        info = entry.stat()
        print(info.st_mtime)

"""
1548163662.3952665
1548163689.1982062
1548163697.9175904
1548163721.1841028
1548163740.765162
1548163769.4702623
"""

os.scandir() 返回一個 ScandirIterator 對象。ScandirIterator 對象中的每一項有 .stat() 方法能獲取關於它指向文件或目錄的信息。.stat() 提供了例如文件大小和最後修改時間的信息。在上面的示例中,代碼打印了 st_time 屬性,該屬性是上次修改文件內容的時間。

pathlib 模塊具有相應的方法,用於獲取相同結果的文件信息:

from pathlib import Path

basepath = Path('my_directory')
for entry in basepath.iterdir():
    info = entry.stat()
    print(info.st_mtime)

"""
1548163662.3952665
1548163689.1982062
1548163697.9175904
1548163721.1841028
1548163740.765162
1548163769.4702623
"""

在上面的例子中,循環 .iterdir() 返回的迭代器並通過對其中每一項調用 .stat() 來獲取文件屬性。st_mtime 屬性是一個浮點類型的值,表示的是時間戳。爲了讓 st_time 返回的值更容易閱讀,你可以編寫一個輔助函數將其轉換爲一個 datetime 對象:

import datetime                                                                   
from pathlib import Path                                                          

def timestamp2datetime(timestamp, convert_to_local=True, utc=8, is_remove_ms=True)
    """                                                                           
    轉換 UNIX 時間戳爲 datetime對象                                                       
    :param timestamp: 時間戳                                                         
    :param convert_to_local: 是否轉爲本地時間                                             
    :param utc: 時區信息,中國爲utc+8                                                     
    :param is_remove_ms: 是否去除毫秒                                                   
    :return: datetime 對象                                                          
    """                                                                           
    if is_remove_ms:                                                              
        timestamp = int(timestamp)                                                
    dt = datetime.datetime.utcfromtimestamp(timestamp)                            
    if convert_to_local:                                                          
        dt = dt + datetime.timedelta(hours=utc)                                   
    return dt                                                                     

def convert_date(timestamp, format='%Y-%m-%d %H:%M:%S'):                          
    dt = timestamp2datetime(timestamp)                                            
    return dt.strftime(format)                                                    

basepath = Path('my_directory')                                                   
for entry in basepath.iterdir():
    if entry.is_file()
    	info = entry.stat()                                                           
    	print('{} 上次修改時間爲 {}'.format(entry.name, timestamp2datetime(info.st_mtime)))  

首先得到 my_directory 中文件的列表以及它們的屬性,然後調用 convert_date() 來轉換文件最後修改時間讓其以一種人類可讀的方式顯示。convert_date() 使用 .strftime() 將datetime類型轉換爲字符串。

上述代碼的輸出結果:

file3.txt 上次修改時間爲 2019-01-24 09:04:39
file2.csv 上次修改時間爲 2019-01-24 09:04:39
file1.py 上次修改時間爲 2019-01-24 09:04:39

將日期和時間轉換爲字符串的語法可能會讓你感到混亂。


創建目錄

你編寫的程序遲早需要創建目錄以便在其中存儲數據。 ospathlib 包含了創建目錄的函數。我們將會考慮如下方法:

方法 描述
os.mkdir() 創建單個子目錄
os.makedirs() 創建多個目錄,包括中間目錄
Pathlib.Path.mkdir() 創建單個或多個目錄

創建單個目錄

要創建單個目錄,把目錄路徑作爲參數傳給 os.mkdir() :

import os

os.mkdir('example_directory')

如果該目錄已經存在,os.mkdir() 將拋出 FileExistsError 異常。或者,你也可以使用 pathlib 來創建目錄:

from pathlib import Path

p = Path('example_directory')
p.mkdir()

如果路徑已經存在,mkdir() 會拋出 FileExistsError 異常:

FileExistsError: [Errno 17] File exists: 'example_directory'

爲了避免像這樣的錯誤拋出, 當發生錯誤時捕獲錯誤並讓你的用戶知道:

from pathlib import Path

p = Path('example_directory')
try:
    p.mkdir()
except FileExistsError as e:
    print(e)

或者,你可以給 .mkdir() 傳入 exist_ok=True 參數來忽略 FileExistsError 異常:

from pathlib import Path

p = Path('example_directory')
p.mkdir(exist_ok=True)

如果目錄已存在,則不會引起錯誤。

創建多個目錄

os.makedirs()os.mkdir() 類似。兩者之間的區別在於,os.makedirs() 不僅可以創建單獨的目錄,還可以遞歸的創建目錄樹。換句話說,它可以創建任何必要的中間文件夾,來確保存在完整的路徑。

os.makedirs() 和在bash中運行 mkdir -p 類似。例如,要創建一組目錄像 2018/10/05,你可以像下面那樣操作:

import os

os.makedirs('2018/10/05', mode=0o770)

上述代碼創建了 2018/10/05 的目錄結構併爲所有者和組用戶提供讀、寫和執行權限。默認的模式爲 0o777 ,增加了其他用戶組的權限。

運行 tree 命令確認我們應用的權限:

$ tree -p -i .
.
[drwxrwx---]  2018
[drwxrwx---]  10
[drwxrwx---]  05

上述代碼打印出當前目錄的目錄樹。 tree 通常被用來以樹形結構列出目錄的內容。傳入 -p-i 參數則會以垂直列表打印出目錄名稱以及其文件權限信息。-p 用於輸出文件權限,-i 則用於讓 tree 命令產生一個沒有縮進線的垂直列表。

正如你所看到的,所有的目錄都擁有 770 權限。另一個方式創建多個目錄是使用 pathlib.Path.mkdir() :

from pathlib import Path

p = Path('2018/10/05')
p.mkdir(parents=True, exist_ok=True)

通過給 Path.mkdir() 傳遞 parents=True 關鍵字參數使它創建 05 目錄和使其路徑有效的所有父級目錄。

在默認情況下,os.makedirs()pathlib.Path.mkdir() 會在目標目錄存在的時候拋出 OSError 。通過每次調用函數時傳遞 exist_ok=True 作爲關鍵字參數則可以覆蓋此行爲(從Python3.2開始)。

運行上述代碼會得到像下面的結構:

└── 2018
    └── 10
        └── 05

我更喜歡在創建目錄時使用 pathlib ,因爲我可以使用相同的函數方法來創建一個或多個目錄。


文件名模式匹配

使用上述方法之一獲取目錄中的文件列表後,你可能希望搜索和特定的模式匹配的文件。

下面這些是你可以使用的方法和函數:

  • endswith()startswith() 字符串方法
  • fnmatch.fnmatch()
  • glob.glob()
  • pathlib.Path.glob()

這些方法和函數是下面要討論的。本小節的示例將在名爲 some_directory 的目錄下執行,該目錄具有以下的結構:

.
├── admin.py
├── data_01_backup.txt
├── data_01.txt
├── data_02_backup.txt
├── data_02.txt
├── data_03_backup.txt
├── data_03.txt
├── sub_dir
│   ├── file1.py
│   └── file2.py
└── tests.py

如果你正在使用 Bash shell,你可以使用以下的命令創建上述目錄結構:

mkdir some_directory
cd some_directory
mkdir sub_dir
touch sub_dir/file1.py sub_dir/file2.py
touch data_{01..03}.txt data_{01..03}_backup.txt admin.py tests.py

這將會創建 some_directory 目錄並進入它,接着創建 sub_dir 。下一行在 sub_dir 創建 file1.pyfile2.py ,最後一行使用擴展創建其它所有文件。

使用字符串方法

Python有幾個內置 修改和操作字符串 的方法。當在匹配文件名時,其中的兩個方法 .startswith().endswith() 非常有用。要做到這點,首先要獲取一個目錄列表,然後遍歷。

import os

for f_name in os.listdir('some_directory'):
    if f_name.endswith('.txt'):
        print(f_name)

上述代碼找到 some_directory 中的所有文件,遍歷並使用 .endswith() 來打印所有擴展名爲 .txt 的文件名。運行代碼在我的電腦上輸出如下:

data_01.txt
data_01_backup.txt
data_02.txt
data_02_backup.txt
data_03.txt
data_03_backup.txt

使用 fnmatch 進行簡單文件名模式匹配

字符串方法匹配的能力是有限的。fnmatch 有對於模式匹配有更先進的函數和方法。我們將考慮使用 fnmatch.fnmatch() ,這是一個支持使用 *? 等通配符的函數。例如,使用 fnmatch 查找目錄中所有 .txt 文件,你可以這樣做:

import os
import fnmatch

for f_name in os.listdir('some_directory'):
    if fnmatch.fnmatch(f_name, '*.txt'):
        print(f_name)

迭代 some_directory 中的文件列表,並使用 .fnmatch() 對擴展名爲 .txt 的文件執行通配符搜索。

更先進的模式匹配

假設你想要查找符合特定掉件的 .txt 文件。例如,你可能指向找到包含單次 data.txt文件,一組下劃線之間的數字,以及文件名中包含單詞 backup 。就類似於 data_01_backup, data_02_backup, 或 data_03_backup

你可以這樣使用 fnmatch.fnmatch() :

import os
import fnmatch

for f_name in os.listdir('some_directory'):
    if fnmatch.fnmatch(f_name, 'data_*_backup.txt'):
        print(f_name)

這裏就僅僅打印出匹配 data_*_backup.txt 模式的文件名稱。模式中的 * 將匹配任何字符,因此運行這段代碼則將查找文件名以 data 開頭並以 backup.txt 的所有文本文件,就行下面的輸出所示 :

data_01_backup.txt
data_02_backup.txt
data_03_backup.txt

使用 glob 進行文件名模式匹配

另一個有用的模式匹配模塊是 glob

.glob()glob 模塊中的左右就像 fnmatch.fnmatch(),但是與 fnmach.fnmatch() 不同的是,它將以 . 開頭的文件視爲特殊文件。

UNIX和相關係統在文件列表中使用通配符像 ?* 表示全匹配。

例如,在UNIX shell中使用 mv *.py python_files 移動所有 .py 擴展名 的文件從當前目錄到 python_files 。這 * 是一個通配符表示任意數量的字符,*.py 是一個全模式。Windows操作系統中不提供此shell功能。但 glob 模塊在Python中添加了此功能,使得Windows程序可以使用這個特性。

這裏有一個使用 glob 模塊在當前目錄下查詢所有Python代碼文件:

import glob

print(glob.glob('*.py'))

glob.glob('*.py') 搜索當前目錄中具有 .py 擴展名的文件,並且將它們以列表的形式返回。 glob 還支持 shell 樣式的通配符來進行匹配 :

import glob

for name in glob.glob('*[0-9]*.txt'):
    print(name)

這將找到所有文件名中包含數字的文本文件(.txt) :

data_01.txt
data_01_backup.txt
data_02.txt
data_02_backup.txt
data_03.txt
data_03_backup.txt

glob 也很容易在子目錄中遞歸的搜索文件:

import glob

for name in glob.iglob('**/*.py', recursive=True):
    print(name)

這裏例子使用了 glob.iglob() 在當前目錄和子目錄中搜索所有的 .py 文件。傳遞 recursive=True 作爲 .iglob() 的參數使其搜索當前目錄和子目錄中的 .py 文件。glob.glob()glob.iglob() 不同之處在於,iglob() 返回一個迭代器而不是一個列表。

運行上述代碼會得到以下結果:

admin.py
tests.py
sub_dir/file1.py
sub_dir/file2.py

pathlib 也包含類似的方法來靈活的獲取文件列表。下面的例子展示了你可以使用 .Path.glob() 列出以字母 p 開始的文件類型的文件列表。

from pathlib import Path

p = Path('.')

for name in p.glob('*.p*'):
    print(name)

調用 p.glob('*.p*') 會返回一個指向當前目錄中所有擴展名以字母 p 開頭的文件的生成器對象。

Path.glob() 和上面討論過的 os.glob() 類似。正如你看到的, pathlib 混合了許多 osos.pathglob 模塊的最佳特性到一個模塊中,這使得使用起來很方便。

回顧一下,這是我們在本節中介紹的功能表:

函數 描述
startswith() 測試一個字符串是否以一個特定的模式開始,返回 True 或 False
endswith() 測試一個字符串是否以一個特定的模式結束,返回 True 或 False
fnmatch.fnmatch(filename, pattern) 測試文件名是否匹配這個模式,返回 True 或 False
glob.glob() 返回一個匹配該模式的文件名列表
pathlib.Path.glob() 返回一個匹配該模式的生成器對象

遍歷目錄和處理文件

一個常見的編程任務是遍歷目錄樹並處理目錄樹中的文件。讓我們來探討一下如何使用內置的Python函數 os.walk() 來實現這一功能。os.walk() 用於通過從上到下或從下到上遍歷樹來生成目錄樹中的文件名。處於本節的目的,我們想操作以下的目錄樹:

├── folder_1
│   ├── file1.py
│   ├── file2.py
│   └── file3.py
├── folder_2
│   ├── file4.py
│   ├── file5.py
│   └── file6.py
├── test1.txt
└── test2.txt

以下是一個示例,演示如何使用 os.walk() 列出目錄樹中的所有文件和目錄。

os.walk() 默認是從上到下遍歷目錄:

import os
for dirpath, dirname, files in os.walk('.'):
   print(f'Found directory: {dirpath}')
   for file_name in files:
       print(file_name)

os.walk() 在每個循環中返回三個值:

  1. 當前文件夾的名稱
  2. 當前文件夾中子文件夾的列表
  3. 當前文件夾中文件的列表

在每次迭代中,會打印出它找到的子目錄和文件的名稱:

Found directory: .
test1.txt
test2.txt
Found directory: ./folder_1
file1.py
file3.py
file2.py
Found directory: ./folder_2
file4.py
file5.py
file6.py

要以自下而上的方式遍歷目錄樹,則將 topdown=False 關鍵字參數傳遞給 os.walk()

for dirpath, dirnames, files in os.walk('.', topdown=False):
    print(f'Found directory: {dirpath}')
    for file_name in files:
        print(file_name)

傳遞 topdown=False 參數將使 os.walk() 首先打印出它在子目錄中找到的文件:

在學習過程中有什麼不懂得可以加我的
python學習qun,855408893
羣裏有不錯的學習視頻教程、開發工具與電子書籍。  
與你分享python企業當下人才需求及怎麼從零基礎學習好python,和學習什麼內容

Found directory: ./folder_1
file1.py
file3.py
file2.py
Found directory: ./folder_2
file4.py
file5.py
file6.py
Found directory: .
test1.txt
test2.txt

如你看見的,程序在列出根目錄的內容之前列出子目錄的內容。 這在在你想要遞歸刪除文件和目錄的情況下非常有用。 你將在以下部分中學習如何執行此操作。 默認情況下,os.walk 不會訪問通過軟連接創建的目錄。 可以通過使用 followlinks = True 參數來覆蓋默認行爲。


創建臨時文件和目錄

Python提供了 tempfile 模塊來便捷的創建臨時文件和目錄。

tempfile 可以在你程序運行時打開並存儲臨時的數據在文件或目錄中。 tempfile 會在你程序停止運行後刪除這些臨時文件。

現在,讓我們看看如何創建一個臨時文件:

from tempfile import  TemporaryFile

# 創建一個臨時文件併爲其寫入一些數據
fp = TemporaryFile('w+t')
fp.write('Hello World!')
# 回到開始,從文件中讀取數據
fp.seek(0)
data = fp.read()
print(data)
# 關閉文件,之後他將會被刪除
fp.close()

第一步是從 tempfile 模塊導入 TemporaryFile 。 接下來,使用 TemporaryFile() 方法並傳入一個你想打開這個文件的模式來創建一個類似於對象的文件。這將創建並打開一個可用作臨時存儲區域的文件。

在上面的示例中,模式爲 w + t,這使得 tempfile 在寫入模式下創建臨時文本文件。 沒有必要爲臨時文件提供文件名,因爲在腳本運行完畢後它將被銷燬。

寫入文件後,您可以從中讀取並在完成處理後將其關閉。 一旦文件關閉後,將從文件系統中刪除。 如果需要命名使用 tempfile 生成的臨時文件,請使用 tempfile.NamedTemporaryFile()

使用 tempfile 創建的臨時文件和目錄存儲在用於存儲臨時文件的特殊系統目錄中。 Python將在目錄列表搜索用戶可以在其中創建文件的目錄。

在Windows上,目錄按順序爲 C:\TEMPC:\TMP\TEMP\TMP。 在所有其他平臺上,目錄按順序爲 / tmp/var/tmp/usr/tmp 。 如果上述目錄中都沒有,tempfile 將在當前目錄中存儲臨時文件和目錄。

.TemporaryFile() 也是一個上下文管理器,因此它可以與with語句一起使用。 使用上下文管理器會在讀取文件後自動關閉和刪除文件:

with TemporaryFile('w+t') as fp:
    fp.write('Hello universe!')
    fp.seek(0)
    fp.read()
# 臨時文件現在已經被關閉和刪除

這將創建一個臨時文件並從中讀取數據。 一旦讀取文件的內容,就會關閉臨時文件並從文件系統中刪除。

tempfile 也可用於創建臨時目錄。 讓我們看一下如何使用 tempfile.TemporaryDirectory()來做到這一點:

import tempfile
import os

tmp = ''
with tempfile.TemporaryDirectory() as tmpdir:
    print('Created temporary directory ', tmpdir)
    tmp = tmpdir
    print(os.path.exists(tmpdir))

print(tmp)
print(os.path.exists(tmp))

調用 tempfile.TemporaryDirectory() 會在文件系統中創建一個臨時目錄,並返回一個表示該目錄的對象。 在上面的示例中,使用上下文管理器創建目錄,目錄的名稱存儲在 tmpdir 變量中。 第三行打印出臨時目錄的名稱,os.path.exists(tmpdir) 來確認目錄是否實際在文件系統中創建。

在上下文管理器退出上下文後,臨時目錄將被刪除,並且對 os.path.exists(tmpdir)的調用將返回False,這意味着該目錄已成功刪除。


刪除文件和目錄

您可以使用 osshutilpathlib 模塊中的方法刪除單個文件,目錄和整個目錄樹。 以下將介紹如何刪除你不再需要的文件和目錄。

Python中刪除文件

要刪除單個文件,請使用 pathlib.Path.unlink()os.remove()os.unlink()

os.remove()os.unlink() 在語義上是相同的。 要使用 os.remove()刪除文件,請執行以下操作:

import os

data_file = 'C:\\Users\\vuyisile\\Desktop\\Test\\data.txt'
os.remove(data_file)

使用 os.unlink() 刪除文件與使用 os.remove() 的方式類似:

import os

data_file = 'C:\\Users\\vuyisile\\Desktop\\Test\\data.txt'
os.unlink(data_file)

在文件上調用 .unlink().remove() 會從文件系統中刪除該文件。 如果傳遞給它們的路徑指向目錄而不是文件,這兩個函數將拋出 OSError 。 爲避免這種情況,可以檢查你要刪除的內容是否是文件,並在確認是文件時執行刪除操作,或者可以使用異常處理來處理 OSError

import os

data_file = 'home/data.txt'
# 如果類型是文件則進行刪除
if os.path.is_file(data_file):
    os.remove(data_file)
else:
    print(f'Error: {data_file} not a valid filename')

os.path.is_file() 檢查 data_file 是否實際上是一個文件。 如果是,則通過調用 os.remove() 刪除它。 如果 data_file 指向文件夾,則會向控制檯輸出錯誤消息。

以下示例說明如何在刪除文件時使用異常處理來處理錯誤:

import os

data_file = 'home/data.txt'
# 使用異常處理
try:
    os.remove(data_file)
except OSError as e:
    print(f'Error: {data_file} : {e.strerror}')

上面的代碼嘗試在檢查其類型之前先刪除該文件。 如果 data_file 實際上不是文件,則拋出的 OSError 將在except子句中處理,並向控制檯輸出錯誤消息。

最後,你還可以使用 pathlib.Path.unlink() 刪除文件:

from pathlib import Path

data_file = Path('home/data.txt')
try:
    data_file.unlink()
except IsADirectoryError as e:
    print(f'Error: {data_file} : {e.strerror}')

這將創建一個名爲 data_filePath 對象,該對象指向一個文件。 在 data_file 上調用.unlink()將刪除 home / data.txt 。 如果 data_file 指向目錄,則引發 IsADirectoryError 。 值得注意的是,上面的Python程序和運行它的用戶具有相同的權限。 如果用戶沒有刪除文件的權限,則會引發 PermissionError

刪除目錄

標準庫提供了一下函數來刪除目錄:

  • os.rmdir()
  • pathlib.Path.rmdir()
  • shutil.rmtree()

要刪除單個目錄或文件夾可以使用 os.rmdir()pathlib.Path.rmdir() 。這兩個函數只在你刪除空目錄的時候有效。如果目錄不爲空,則會拋出 OSError 。下面演示如何刪除一個文件夾:

import os

trash_dir = 'my_documents/bad_dir'

try:
    os.rmdir(trash_dir)
except OSError as e:
    print(f'Error: {trash_dir} : {e.strerror}')

現在,trash_dir 已經通過 os.rmdir() 被刪除了。如果目錄不爲空,則會在屏幕上打印錯誤信息:

Traceback (most recent call last):
  File '<stdin>', line 1, in <module>
OSError: [Errno 39] Directory not empty: 'my_documents/bad_dir'

同樣,你也可使用 pathlib 來刪除目錄:

from pathlib import Path

trash_dir = Path('my_documents/bad_dir')

try:
    trash_dir.rmdir()
except OSError as e:
    print(f'Error: {trash_dir} : {e.strerror}')

這裏創建了一個 Path 對象指向要被刪除的目錄。如果目錄爲空,調用 Path 對象的 .rmdir() 方法刪除它。

刪除完整的目錄樹

要刪除非空目錄和完整的目錄樹,Python提供了 shutil.rmtree() :

import shutil

trash_dir = 'my_documents/bad_dir'

try:
    shutil.rmtree(trash_dir)
except OSError as e:
    print(f'Error: {trash_dir} : {e.strerror}')

當調用 shutil.rmtree() 時,trash_dir 中的所有內容都將被刪除。 在某些情況下,你可能希望以遞歸方式刪除空文件夾。 你可以使用上面討論的方法之一結合 os.walk() 來完成此操作:

import os

for dirpath, dirnames, files in os.walk('.', topdown=False):
    try:
        os.rmdir(dirpath)
    except OSError as ex:
        pass

這將遍歷目錄樹並嘗試刪除它找到的每個目錄。 如果目錄不爲空,則引發OSError並跳過該目錄。 下表列出了本節中涉及的功能:

函數 描述
os.remove() 刪除單個文件,不能刪除目錄
os.unlink() 和os.remove()一樣,職能刪除單個文件
pathlib.Path.unlink() 刪除單個文件,不能刪除目錄
os.rmdir() 刪除一個空目錄
pathlib.Path.rmdir() 刪除一個空目錄
shutil.rmtree() 刪除完整的目錄樹,可用於刪除非空目錄

複製、移動和重命名文件和目錄

Python附帶了 shutil 模塊。 shutil 是shell實用程序的縮寫。 它爲文件提供了許多高級操作,來支持文件和目錄的複製,歸檔和刪除。 在本節中,你將學習如何移動和複製文件和目錄。

複製文件

shutil 提供了一些複製文件的函數。 最常用的函數是 shutil.copy()shutil.copy2() 。 使用shutil.copy() 將文件從一個位置複製到另一個位置,請執行以下操作:

import shutil

src = 'path/to/file.txt'
dst = 'path/to/dest_dir'
shutil.copy(src, dst)

shutil.copy() 與基於UNIX的系統中的 cp 命令相當。 shutil.copy(src,dst) 會將文件 src 複製到 dst 中指定的位置。 如果 dst 是文件,則該文件的內容將替換爲 src 的內容。 如果 dst 是目錄,則 src 將被複制到該目錄中。 shutil.copy() 僅複製文件的內容和文件的權限。 其他元數據(如文件的創建和修改時間)不會保留。

要在複製時保留所有文件元數據,請使用 shutil.copy2()

import shutil

src = 'path/to/file.txt'
dst = 'path/to/dest_dir'
shutil.copy2(src, dst)

使用 .copy2() 保留有關文件的詳細信息,例如上次訪問時間,權限位,上次修改時間和標誌。

複製目錄

雖然 shutil.copy() 只複製單個文件,但 shutil.copytree() 將複製整個目錄及其中包含的所有內容。 shutil.copytree(src,dest) 接收兩個參數:源目錄和將文件和文件夾複製到的目標目錄。

以下是如何將一個文件夾的內容複製到其他位置的示例:

import shutil
dst = shutil.copytree('data_1', 'data1_backup')
print(dst)  # data1_backup

在此示例中,.copytree()data_1 的內容複製到新位置 data1_backup 並返回目標目錄。 目標目錄不能是已存在的。 它將被創建而不帶有其父目錄。 shutil.copytree() 是備份文件的一個好方法。

移動文件和目錄

要將文件或目錄移動到其他位置,請使用 shutil.move(src,dst)

src 是要移動的文件或目錄,dst 是目標:

import shutil
dst = shutil.move('dir_1/', 'backup/')
print(dst)  # 'backup'

如果 backup/ 存在,則 shutil.move('dir_1/','backup/')dir_1/ 移動到 backup/ 。 如果 backup/ 不存在,則 dir_1/ 將重命名爲 backup

重命名文件和目錄

Python包含用於重命名文件和目錄的 os.rename(src,dst)

import os
os.rename('first.zip', 'first_01.zip')

上面的行將 first.zip 重命名爲 first_01.zip 。 如果目標路徑指向目錄,則會拋出 OSError

重命名文件或目錄的另一種方法是使用 pathlib 模塊中的 rename()

from pathlib import Path
data_file = Path('data_01.txt')
data_file.rename('data.txt')

要使用 pathlib 重命名文件,首先要創建一個 pathlib.Path() 對象,該對象包含要替換的文件的路徑。 下一步是在路徑對象上調用 rename() 並傳入你要重命名的文件或目錄的新名稱。


歸檔

歸檔是將多個文件打包成一個文件的便捷方式。 兩種最常見的存檔類型是ZIP和TAR。 你編寫的Python程序可以創建存檔文件,讀取存檔文件和從存檔文件中提取數據。 你將在本節中學習如何讀取和寫入兩種壓縮格式。

讀取ZIP文件

zipfile 模塊是一個底層模塊,是Python標準庫的一部分。 zipfile 具有可以輕鬆打開和提取ZIP文件的函數。 要讀取ZIP文件的內容,首先要做的是創建一個 ZipFile 對象。ZipFile 對象類似於使用 open() 創建的文件對象。ZipFile 也是一個上下文管理器,因此支持with語句:

import zipfile

with zipfile.ZipFile('data.zip', 'r') as zipobj:
    pass

這裏創建一個 ZipFile 對象,傳入ZIP文件的名稱並以讀取模式下打開。 打開ZIP文件後,可以通過 zipfile 模塊提供的函數訪問有關存檔文件的信息。 上面示例中的 data.zip 存檔是從名爲 data 的目錄創建的,該目錄包含總共5個文件和1個子目錄:

.
|
├── sub_dir/
|   ├── bar.py
|   └── foo.py
|
├── file1.py
├── file2.py
└── file3.py

要獲取存檔文件中的文件列表,請在 ZipFile 對象上調用 namelist()

import zipfile

with zipfile.ZipFile('data.zip', 'r') as zipobj:
    zipobj.namelist()

這會生成一個文件列表:

['file1.py', 'file2.py', 'file3.py', 'sub_dir/', 'sub_dir/bar.py', 'sub_dir/foo.py']

.namelist() 返回存檔文件中文件和目錄的名稱列表。要檢索有關存檔文件中文件的信息,使用 .getinfo()

import zipfile

with zipfile.ZipFile('data.zip', 'r') as zipobj:
    bar_info = zipobj.getinfo('sub_dir/bar.py')
    print(bar_info.file_size)

這將輸出:

15277

.getinfo() 返回一個 ZipInfo 對象,該對象存儲有關存檔文件的單個成員的信息。 要獲取有關存檔文件中文件的信息,請將其路徑作爲參數傳遞給 .getinfo() 。 使用 getinfo() ,你可以檢索有關存檔文件成員的信息,例如上次修改文件的日期,壓縮大小及其完整文件名。 訪問 .file_size 將以字節爲單位檢索文件的原始大小。

以下示例說明如何在Python REPL中檢索有關已歸檔文件的更多詳細信息。 假設已導入 zipfile 模塊,bar_info 與在前面的示例中創建的對象相同:

>>> bar_info.date_time
(2018, 10, 7, 23, 30, 10)
>>> bar_info.compress_size
2856
>>> bar_info.filename
'sub_dir/bar.py'

bar_info 包含有關 bar.py 的詳細信息,例如壓縮的大小及其完整路徑。

第一行顯示瞭如何檢索文件的上次修改日期。 下一行顯示瞭如何在歸檔後獲取文件的大小。 最後一行顯示了存檔文件中 bar.py 的完整路徑。

ZipFile 支持上下文管理器協議,這就是你可以將它與with語句一起使用的原因。 操作完成後會自動關閉 ZipFile 對象。 嘗試從已關閉的 ZipFile 對象中打開或提取文件將導致錯誤。

提取ZIP文件

zipfile 模塊允許你通過 .extract().extractall() 從ZIP文件中提取一個或多個文件。

默認情況下,這些方法將文件提取到當前目錄。 它們都採用可選的路徑參數,允許指定要將文件提取到的其他指定目錄。 如果該目錄不存在,則會自動創建該目錄。 要從壓縮文件中提取文件,請執行以下操作:

>>> import zipfile
>>> import os

>>> os.listdir('.')
['data.zip']

>>> data_zip = zipfile.ZipFile('data.zip', 'r')

>>> # 提取單個文件到當前目錄
>>> data_zip.extract('file1.py')
'/home/test/dir1/zip_extract/file1.py'

>>> os.listdir('.')
['file1.py', 'data.zip']

>>> # 提所有文件到指定目錄
>>> data_zip.extractall(path='extract_dir/')

>>> os.listdir('.')
['file1.py', 'extract_dir', 'data.zip']

>>> os.listdir('extract_dir')
['file1.py', 'file3.py', 'file2.py', 'sub_dir']

>>> data_zip.close()

第三行代碼是對 os.listdir() 的調用,它顯示當前目錄只有一個文件 data.zip

接下來,以讀取模式下打開 data.zip 並調用 .extract() 從中提取 file1.py.extract() 返回提取文件的完整文件路徑。 由於沒有指定路徑,.extract() 會將 file1.py 提取到當前目錄。

下一行打印一個目錄列表,顯示當前目錄現在包括除原始存檔文件之外的存檔文件。 之後顯示瞭如何將整個存檔提取到指定目錄中。.extractall() 創建 extract_dir 並將 data.zip 的內容提取到其中。 最後一行關閉ZIP存檔文件。

從加密的文檔提取數據

zipfile 支持提取受密碼保護的ZIP。 要提取受密碼保護的ZIP文件,請將密碼作爲參數傳遞給 .extract().extractall() 方法:

>>> import zipfile

>>> with zipfile.ZipFile('secret.zip', 'r') as pwd_zip:
...     # 從加密的文檔提取數據
...     pwd_zip.extractall(path='extract_dir', pwd='Quish3@o')

將以讀取模式打開 secret.zip 存檔。 密碼提供給 .extractall() ,並且壓縮文件內容被提取到 extract_dir 。 由於with語句,在完成提取後,存檔文件會自動關閉。

創建新的存檔文件

要創建新的ZIP存檔,請以寫入模式(w)打開 ZipFile 對象並添加要歸檔的文件:

>>> import zipfile

>>> file_list = ['file1.py', 'sub_dir/', 'sub_dir/bar.py', 'sub_dir/foo.py']
>>> with zipfile.ZipFile('new.zip', 'w') as new_zip:
...     for name in file_list:
...         new_zip.write(name)

在該示例中,new_zip 以寫入模式打開,file_list 中的每個文件都添加到存檔文件中。 with語句結束後,將關閉 new_zip 。 以寫入模式打開ZIP文件會刪除壓縮文件的內容並創建新存檔文件。

要將文件添加到現有的存檔文件,請以追加模式打開 ZipFile 對象,然後添加文件:

>>> with zipfile.ZipFile('new.zip', 'a') as new_zip:
...     new_zip.write('data.txt')
...     new_zip.write('latin.txt')

這裏打開在上一個示例中以追加模式創建的 new.zip 存檔。 在追加模式下打開 ZipFile 對象允許將新文件添加到ZIP文件而不刪除其當前內容。 將文件添加到ZIP文件後,with語句將脫離上下文並關閉ZIP文件。

打開TAR存檔文件

TAR文件是像ZIP等未壓縮的文件存檔。 它們可以使用 gzipbzip2lzma 壓縮方法進行壓縮。 TarFile 類允許讀取和寫入TAR存檔。

下面是從存檔中讀取:

import tarfile

with tarfile.open('example.tar', 'r') as tar_file:
    print(tar_file.getnames())

tarfile 對象像大多數類似文件的對象一樣打開。 它們有一個 open() 函數,它採用一種模式來確定文件的打開方式。

使用“r”,“w”或“a”模式分別打開未壓縮的TAR文件以進行讀取,寫入和追加。 要打開壓縮的TAR文件,請將模式參數傳遞給 tarfile.open(),其格式爲 filemode [:compression] 。 下表列出了可以打開TAR文件的可能模式:

模式 行爲
r 以無壓縮的讀取模式打開存檔
r:gz 以gzip壓縮的讀取模式打開存檔
r:bz2 以bzip2壓縮的讀取模式打開存檔
w 以無壓縮的寫入模式打開存檔
w:gz 以gzip壓縮的寫入模式打開存檔
w:xz 以lzma壓縮的寫入模式打開存檔
a 以無壓縮的追加模式打開存檔

.open() 默認爲’r’模式。 要讀取未壓縮的TAR文件並檢索其中的文件名,請使用 .getnames()

>>> import tarfile

>>> tar = tarfile.open('example.tar', mode='r')
>>> tar.getnames()
['CONTRIBUTING.rst', 'README.md', 'app.py']

這以列表的方式返回存檔中內容的名字。

注意:爲了向你展示如何使用不同的tarfile對象方法,示例中的TAR文件在交互式REPL會話中手動打開和關閉。

通過這種方式與TAR文件交互,你可以查看運行每個命令的輸出。 通常,你可能希望使用上下文管理器來打開類似文件的對象。

此外可以使用特殊屬性訪問存檔中每個條目的元數據:

>>> for entry in tar.getmembers():
...     print(entry.name)
...     print(' Modified:', time.ctime(entry.mtime))
...     print(' Size    :', entry.size, 'bytes')
...     print()
CONTRIBUTING.rst
 Modified: Sat Nov  1 09:09:51 2018
 Size    : 402 bytes

README.md
 Modified: Sat Nov  3 07:29:40 2018
 Size    : 5426 bytes

app.py
 Modified: Sat Nov  3 07:29:13 2018
 Size    : 6218 bytes

在此示例中,循環遍歷 .getmembers() 返回的文件列表,並打印出每個文件的屬性。.getmembers() 返回的對象具有可以通過編程方式訪問的屬性,例如歸檔中每個文件的名稱,大小和上次修改時間。 在讀取或寫入存檔後,必須關閉它以釋放系統資源。

從TAR存檔中提取文件

在本節中,你將學習如何使用以下方法從TAR存檔中提取文件:

  • .extract()
  • .extractfile()
  • .extractall()

要從TAR存檔中提取單個文件,請使用 extract() ,傳入文件名:

>>> tar.extract('README.md')
>>> os.listdir('.')
['README.md', 'example.tar']

README.md 文件從存檔中提取到文件系統。 調用 os.listdir() 確認 README.md 文件已成功提取到當前目錄中。 要從存檔中解壓縮或提取所有內容,請使用 .extractall()

>>> tar.extractall(path="extracted/")

.extractall() 有一個可選的 path 參數來指定解壓縮文件的去向。 這裏,存檔被提取到 extracted 目錄中。 以下命令顯示已成功提取存檔:

$ ls
example.tar  extracted  README.md

$ tree
.
├── example.tar
├── extracted
|   ├── app.py
|   ├── CONTRIBUTING.rst
|   └── README.md
└── README.md

1 directory, 5 files

$ ls extracted/
app.py  CONTRIBUTING.rst  README.md

要提取文件對象以進行讀取或寫入,請使用 .extractfile() ,它接收 文件名或 TarInfo 對象作爲參數。 .extractfile() 返回一個可以讀取和使用的類文件對象:

>>> f = tar.extractfile('app.py')
>>> f.read()
>>> tar.close()

打開的存檔應在讀取或寫入後始終關閉。 要關閉存檔,請在存檔文件句柄上調用 .close() ,或在創建 tarfile對象時使用with語句,以便在完成後自動關閉存檔。 這將釋放系統資源,並將你對存檔所做的任何更改寫入文件系統。

創建新的TAR存檔

創建新的TAR存檔,你可以這樣操作:

>>> import tarfile

>>> file_list = ['app.py', 'config.py', 'CONTRIBUTORS.md', 'tests.py']
>>> with tarfile.open('packages.tar', mode='w') as tar:
...     for file in file_list:
...         tar.add(file)

>>> # Read the contents of the newly created archive
>>> with tarfile.open('package.tar', mode='r') as t:
...     for member in t.getmembers():
...         print(member.name)
app.py
config.py
CONTRIBUTORS.md
tests.py

首先,你要創建要添加到存檔的文件列表,這樣你就不必手動添加每個文件。

下一行使用with光線文管理器在寫入模式下打開名爲 packages.tar 的新存檔。 以寫入模式(‘w’)打開存檔使你可以將新文件寫入存檔。 將刪除存檔中的所有現有文件,並創建新存檔。

創建並填充存檔後,with上下文管理器會自動關閉它並將其保存到文件系統。 最後三行打開剛剛創建的存檔,並打印出其中包含的文件的名稱。

要將新文件添加到現有存檔,請以追加模式(‘a’)打開存檔:

>>> with tarfile.open('package.tar', mode='a') as tar:
...     tar.add('foo.bar')

>>> with tarfile.open('package.tar', mode='r') as tar:
...     for member in tar.getmembers():
...         print(member.name)
app.py
config.py
CONTRIBUTORS.md
tests.py
foo.bar

在追加模式下打開存檔允許你向其添加新文件而不刪除其中已存在的文件。

使用壓縮存檔

tarfile 可以讀取和寫入使用 gzipbzip2lzma 壓縮的TAR存檔文件。 要讀取或寫入壓縮存檔,請使用tarfile.open() ,爲壓縮類型傳遞適當的模式。

例如,要讀取或寫入使用 gzip 壓縮的TAR存檔的數據,請分別使用 'r:gz''w:gz' 模式:

>>> files = ['app.py', 'config.py', 'tests.py']
>>> with tarfile.open('packages.tar.gz', mode='w:gz') as tar:
...     tar.add('app.py')
...     tar.add('config.py')
...     tar.add('tests.py')

>>> with tarfile.open('packages.tar.gz', mode='r:gz') as t:
...     for member in t.getmembers():
...         print(member.name)
app.py
config.py
tests.py

'w:gz' 以寫模式模式打開 gzip 壓縮的存檔,'r:gz' 以讀模式打開 gzip 壓縮的存檔。 無法在追加模式下打開壓縮存檔。 要將文件添加到壓縮存檔,你必須創建新存檔。


一個更簡單的方式創建存檔

Python標準庫還支持使用 shutil 模塊中的高級方法創建TAR和ZIP存檔。 shutil 中的歸檔實用工具允許你創建,讀取和提取ZIP和TAR歸檔。 這些實用工具依賴於較底層的 tarfilezipfile 模塊。

使用 shutil.make_archive() 創建存檔

shutil.make_archive() 至少接收兩個參數:歸檔的名稱和歸檔格式。

默認情況下,它將當前目錄中的所有文件壓縮爲 format 參數中指定的歸檔格式。 你可以傳入可選的 root_dir 參數來壓縮不同目錄中的文件。 .make_archive() 支持 ziptarbztargztar 存檔格式。

以下是使用 shutil 創建TAR存檔的方法:

import shutil

# shutil.make_archive(base_name, format, root_dir)
shutil.make_archive('data/backup', 'tar', 'data/')

這將複製 data / 中的所有內容,並在文件系統中創建名爲 backup.tar 的存檔並返回其名稱。 要提取存檔,請調用 .unpack_archive()

shutil.unpack_archive('backup.tar', 'extract_dir/')

調用 .unpack_archive() 並傳入存檔名稱和目標目錄,將 backup.tar 的內容提取到 extract_dir/ 中。 ZIP存檔可以以相同的方式創建和提取。


讀取多個文件

Python支持通過 fileinput 模塊從多個輸入流或文件列表中讀取數據。 此模塊允許你快速輕鬆地循環遍歷一個或多個文本文件的內容。 以下是使用 fileinput 的典型方法:

import fileinput
for line in fileinput.input()
    process(line)

fileinput 默認從傳遞給 sys.argv 的命令行參數獲取其輸入。

使用 fileinput 循環遍歷多個文件

讓我們使用 fileinput 構建一個普通的UNIX工具 cat 的原始版本。 cat 工具按順序讀取文件,將它們寫入標準輸出。 當在命令行參數中給出多個文件時,cat 將連接文本文件並在終端中顯示結果:

# File: fileinput-example.py
import fileinput
import sys

files = fileinput.input()
for line in files:
    if fileinput.isfirstline():
        print(f'\n--- Reading {fileinput.filename()} ---')
    print(' -> ' + line, end='')
print()

在當前目錄中有兩個文本文件,運行此命令會產生以下輸出:

$ python3 fileinput-example.py bacon.txt cupcake.txt
--- Reading bacon.txt ---
 -> Spicy jalapeno bacon ipsum dolor amet in in aute est qui enim aliquip,
 -> irure cillum drumstick elit.
 -> Doner jowl shank ea exercitation landjaeger incididunt ut porchetta.
 -> Tenderloin bacon aliquip cupidatat chicken chuck quis anim et swine.
 -> Tri-tip doner kevin cillum ham veniam cow hamburger.
 -> Turkey pork loin cupidatat filet mignon capicola brisket cupim ad in.
 -> Ball tip dolor do magna laboris nisi pancetta nostrud doner.

--- Reading cupcake.txt ---
 -> Cupcake ipsum dolor sit amet candy I love cheesecake fruitcake.
 -> Topping muffin cotton candy.
 -> Gummies macaroon jujubes jelly beans marzipan.

fileinput 允許你檢索有關每一行的更多信息,例如它是否是第一行(.isfirstline()),行號(.lineno())和文件名(.filename())。 你可以在 這裏 讀更多關於它的內容。


總結

你現在知道如何使用Python對文件和文件組執行最常見的操作。 你已經瞭解使用不同的內置模塊來讀取,查找和操作文件。

爲解決初學者學習上的困難,專門建立的Python學習扣QUN:⑧⑤⑤-④零⑧-⑧⑨③從零基礎開始到Python各領域的項目實戰教程、開發工具與電子書籍。與你分享企業當下對於python人才需求及學好python的高效技巧,不停更新最新教程!點擊加入我們的 python學習圈

你現在可以用Python來實現:

  • 獲取目錄內容和文件屬性
  • 創建目錄和目錄樹
  • 使用匹配模式匹配文件名
  • 創建臨時文件和目錄
  • 移動,重命名,複製和刪除文件或目錄
  • 從不同類型的存檔文件中讀取和提取數據
  • 使用 fileinput 同時讀取多個文件
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章