有時我們需要在 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 檔案。類源碼為:
[python] view plaincopy
- # 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()