Chinaunix

标题: Python压缩文件夹/解压缩zip文件 [打印本页]

---

作者: usli    时间: 2015-07-03 10:53
标题: Python压缩文件夹/解压缩zip文件
使用python自带的zipfile模块做压缩文件夹/解压缩zip文件功能

[Python]代码

1. #coding=utf-8
   
2. #甄码农python代码
   
3. #使用zipfile做目录压缩，解压缩功能
   
4. 
   
5. import os,os.path
   
6. import zipfile
   
7. 
   
8. def zip_dir(dirname,zipfilename):
   
9.     filelist = []
   
10.     if os.path.isfile(dirname):
    
11.         filelist.append(dirname)
    
12.     else :
    
13.         for root, dirs, files in os.walk(dirname):
    
14.             for name in files:
    
15.                 filelist.append(os.path.join(root, name))
    
16.          
    
17.     zf = zipfile.ZipFile(zipfilename, "w", zipfile.zlib.DEFLATED)
    
18.     for tar in filelist:
    
19.         arcname = tar[len(dirname):]
    
20.         #print arcname
    
21.         zf.write(tar,arcname)
    
22.     zf.close()
    
23. 
    
24. 
    
25. def unzip_file(zipfilename, unziptodir):
    
26.     if not os.path.exists(unziptodir): os.mkdir(unziptodir, 0777)
    
27.     zfobj = zipfile.ZipFile(zipfilename)
    
28.     for name in zfobj.namelist():
    
29.         name = name.replace('\\','/')
    
30.         
    
31.         if name.endswith('/'):
    
32.             os.mkdir(os.path.join(unziptodir, name))
    
33.         else:            
    
34.             ext_filename = os.path.join(unziptodir, name)
    
35.             ext_dir= os.path.dirname(ext_filename)
    
36.             if not os.path.exists(ext_dir) : os.mkdir(ext_dir,0777)
    
37.             outfile = open(ext_filename, 'wb')
    
38.             outfile.write(zfobj.read(name))
    
39.             outfile.close()
    
40. 
    
41. if __name__ == '__main__':
    
42.     zip_dir(r'E:/python/learning',r'E:/python/learning/zip.zip')
    
43.     unzip_file(r'E:/python/learning/zip.zip',r'E:/python/learning2')

复制代码

---

欢迎光临 Chinaunix (http://bbs.chinaunix.net/)

Powered by Discuz! X3.2