python怎樣壓縮和解壓縮ZIP文件(轉)

有時我們需要在 Python 中使用 zip 文件，而在1.6版中，Python 就已經提供了 zipfile 模塊可以進行這樣的操作。不過 Python 中的 zipfile 模塊不能處理多卷的情況，不過這種情況並不多見，因此在通常情況下已經足夠使用了。下面我只是對一些基本的 zipfile 操作進行了記錄，足以應付大部分的情況了。

zipfile 模塊可以讓你打開或寫入一個 zip 文件。比如：

import zipfile

z = zipfile.ZipFile('zipfilename', mode='r')

  這樣就打開了一個 zip 文件，如果mode為'w'或'a'則表示要寫入一個 zip 文件。如果是寫入，則還可以跟上第三個參數：

  compression=zipfile.ZIP_DEFLATED 或

  compression=zipfile.ZIP_STORED ZIP_DEFLATED是壓縮標志，如果使用它需要編譯了zlib模塊。而后一個只是用zip進行打包，並不壓縮。

在打開了zip文件之后就可以根據需要是讀出zip文件的內容還是將內容保存到 zip 文件中。

讀出zip中的內容

很簡單，zipfile 對象提供了一個read(name)的方法。name為 zip文件中的一個文件入口，執行完成之后，將返回讀出的內容，你把它保存到想到的文件中即可。

寫入zip文件

有兩種方式，一種是直接寫入一個已經存在的文件，另一種是寫入一個字符串。

對 於第一種使用 zipfile 對象的 write(filename, arcname, compress_type)，后兩個參數是可以忽略的。第一個參數是文件名，第二個參數是表示在 zip 文件中的名字，如果沒有給出，表示使用與filename一樣的名字。compress_type是壓縮標志，它可以覆蓋創建 zipfile 時的參數。第二種是使用 zipfile 對象的 writestr(zinfo_or_arcname, bytes)，第一個參數是zipinfo 對象或寫到壓縮文件中的壓縮名，第二個參數是字符串。使用這個方法可以動態的組織文件的內容。

需要注意的是在讀出時，因為只能讀出內容，因此如果想實現按目錄結構展開 zip 文件的話，這些操作需要自已來完成，比如創建目錄，創建文件並寫入。而寫入時，則可以根據需要動態組織在 zip 文件中的目錄結構，這樣可以不按照原來的目錄結構來生成 zip 文件。

於是我為了方便使用，創建了自已的一個 ZFile 類，主要是實現象 winrar 的右鍵菜單中的壓縮到的功能--即將一個zip文件壓縮到指定目錄，自動創建相應的子目錄。再有就是方便生成 zip 文件。類源碼為：

# coding:cp936
# Zfile.py
# xxteach.com
import zipfile
import os.path
import os

class ZFile(object):
    def __init__(self, filename, mode='r', basedir=''):
        self.filename = filename
        self.mode = mode
        if self.mode in ('w', 'a'):
            self.zfile = zipfile.ZipFile(filename, self.mode, compression=zipfile.ZIP_DEFLATED)
        else:
            self.zfile = zipfile.ZipFile(filename, self.mode)
        self.basedir = basedir
        if not self.basedir:
            self.basedir = os.path.dirname(filename)

    def addfile(self, path, arcname=None):
        path = path.replace('//', '/')
        if not arcname:
            if path.startswith(self.basedir):
                arcname = path[len(self.basedir):]
            else:
                arcname = ''
        self.zfile.write(path, arcname)

    def addfiles(self, paths):
        for path in paths:
            if isinstance(path, tuple):
                self.addfile(*path)
            else:
                self.addfile(path)

    def close(self):
        self.zfile.close()

    def extract_to(self, path):
        for p in self.zfile.namelist():
            self.extract(p, path)

    def extract(self, filename, path):
        if not filename.endswith('/'):
            f = os.path.join(path, filename)
            dir = os.path.dirname(f)
            if not os.path.exists(dir):
                os.makedirs(dir)
            file(f, 'wb').write(self.zfile.read(filename))


def create(zfile, files):
    z = ZFile(zfile, 'w')
    z.addfiles(files)
    z.close()

def extract(zfile, path):
    z = ZFile(zfile)
    z.extract_to(path)
    z.close()