随笔 - 895  文章 - 0  评论 - 179  0

有时我们需要在 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
  1. # coding:cp936  
  2. # Zfile.py  
  3. # xxteach.com  
  4. import zipfile   
  5. import os.path   
  6. import os   
  7.    
  8. class ZFile(object):   
  9.     def __init__(self, filename, mode='r', basedir=''):   
  10.         self.filename = filename   
  11.         self.mode = mode   
  12.         if self.mode in ('w', 'a'):   
  13.             self.zfile = zipfile.ZipFile(filename, self.mode, compression=zipfile.ZIP_DEFLATED)   
  14.         else:   
  15.             self.zfile = zipfile.ZipFile(filename, self.mode)   
  16.         self.basedir = basedir   
  17.         if not self.basedir:   
  18.             self.basedir = os.path.dirname(filename)   
  19.           
  20.     def addfile(self, path, arcname=None):   
  21.         path = path.replace('//', '/')   
  22.         if not arcname:   
  23.             if path.startswith(self.basedir):   
  24.                 arcname = path[len(self.basedir):]   
  25.             else:   
  26.                 arcname = ''   
  27.         self.zfile.write(path, arcname)   
  28.               
  29.     def addfiles(self, paths):   
  30.         for path in paths:   
  31.             if isinstance(path, tuple):   
  32.                 self.addfile(*path)   
  33.             else:   
  34.                 self.addfile(path)   
  35.               
  36.     def close(self):   
  37.         self.zfile.close()   
  38.           
  39.     def extract_to(self, path):   
  40.         for p in self.zfile.namelist():   
  41.             self.extract(p, path)   
  42.               
  43.     def extract(self, filename, path):   
  44.         if not filename.endswith('/'):   
  45.             f = os.path.join(path, filename)   
  46.             dir = os.path.dirname(f)   
  47.             if not os.path.exists(dir):   
  48.                 os.makedirs(dir)   
  49.             file(f, 'wb').write(self.zfile.read(filename))   
  50.               
  51.           
  52. def create(zfile, files):   
  53.     z = ZFile(zfile, 'w')   
  54.     z.addfiles(files)   
  55.     z.close()   
  56.       
  57. def extract(zfile, path):   
  58.     z = ZFile(zfile)   
  59.     z.extract_to(path)   
  60.     z.close()  

 

posted on 2013-03-27 20:47 一个人的天空@ 阅读(...) 评论(...) 编辑 收藏