5.7 讀寫壓縮文件

問題

你想讀寫一個gzip或bz2格式的壓縮文件。

解決方案

gzip 和 bz2 模塊可以很容易的處理這些文件。兩個模塊都為 open() 函數(shù)提供了另外的實現(xiàn)來解決這個問題。比如，為了以文本形式讀取壓縮文件，可以這樣做：

# gzip compression
import gzip
with gzip.open('somefile.gz', 'rt') as f:
    text = f.read()

# bz2 compression
import bz2
with bz2.open('somefile.bz2', 'rt') as f:
    text = f.read()

類似的，為了寫入壓縮數(shù)據(jù)，可以這樣做：

# gzip compression
import gzip
with gzip.open('somefile.gz', 'wt') as f:
    f.write(text)

# bz2 compression
import bz2
with bz2.open('somefile.bz2', 'wt') as f:
    f.write(text)

如上，所有的I/O操作都使用文本模式并執(zhí)行Unicode的編碼/解碼。類似的，如果你想操作二進制數(shù)據(jù)，使用rb或者wb文件模式即可。

討論

大部分情況下讀寫壓縮數(shù)據(jù)都是很簡單的。但是要注意的是選擇一個正確的文件模式是非常重要的。如果你不指定模式，那么默認的就是二進制模式，如果這時候程序想要接受的是文本數(shù)據(jù)，那么就會出錯。gzip.open() 和 bz2.open() 接受跟內(nèi)置的 open() 函數(shù)一樣的參數(shù)，包括 encoding，errors，newline 等等。

當寫入壓縮數(shù)據(jù)時，可以使用 compresslevel 這個可選的關(guān)鍵字參數(shù)來指定一個壓縮級別。比如：

with gzip.open('somefile.gz', 'wt', compresslevel=5) as f:
    f.write(text)

默認的等級是9，也是最高的壓縮等級。等級越低性能越好，但是數(shù)據(jù)壓縮程度也越低。

最后一點，gzip.open() 和 bz2.open() 還有一個很少被知道的特性，它們可以作用在一個已存在并以二進制模式打開的文件上。比如，下面代碼是可行的：

import gzip
f = open('somefile.gz', 'rb')
with gzip.open(f, 'rt') as g:
    text = g.read()

這樣就允許 gzip 和 bz2 模塊可以工作在許多類文件對象上，比如套接字，管道和內(nèi)存中文件等。