python tarfile模块

wocaosinima

下使用tarfile模块来实现文件归档压缩与解压





1．压缩，创建tar.gz包

#!/usr/bin/env python

import os
import tarfile

#创建压缩包名

tar = tarfile.open("/tmp/tartest.tar.gz","w:gz")

#创建压缩包

for root,dir,files in os.walk("/tmp/tartest"):



        for file in files:
                fullpath = os.path.join(root,file)
                tar.add(fullpath)

tar.close()


2．解压tar.gz包

#!/usr/bin/env python

import tarfile

tar = tarfile.open(“/tmp/tartest.tar.gz”)
names = tar.getnames()

for name in names:
       tar.extract(name,path=”/tmp”)

tar.close()








import zipfile


file = zipfile.ZipFile("samples/sample.zip", "r")     # 打开压缩包


# 列出压缩包内文件名
for name in file.namelist():         
    print name,


# 列出压缩包内文件信息
for info in file.infolist():
    print info.filename, info.date_time, info.file_size


sample.txt sample.jpg
sample.txt (1999, 9, 11, 20, 11, 8) 302
sample.jpg (1999, 9, 18, 16, 9, 44) 4762
---------------------------------------------------------
从 ZIP 文件中读取数据
调用 read 方法就可以从 ZIP 文档中读取数据. 它接受一个文件名作为参数, 返回字符串
---------------------------------------------------------
import zipfile


file = zipfile.ZipFile("samples/sample.zip", "r")   # 打开压缩包


for name in file.namelist():               #列出压缩包内文件名称
    data = file.read(name)                 # 读取文件到data
    print name, len(data), repr(data[:10])        # 输出文件对象数量及内容


sample.txt 302 'We will pe'
sample.jpg 4762 '\377\330\377\340\000\020JFIF'
---------------------------------------------------------
ZIP 文件写入数据
使用 zipfile 模块将文件储存在 ZIP 文件里
向压缩档加入文件很简单, 将文件名, 文件在 ZIP 档中的名称传递给 write 方法即可.
write 方法的第三个可选参数用于控制是否使用压缩.默认为 zipfile.ZIP_STORED , 意味着只是将数据储存在档案里而不进行任何压缩.
如果安装了 zlib 模块, 那么就可以使用 zipfile.ZIP_DEFLATED 进行压缩.
---------------------------------------------------------
import zipfile
import glob, os


file = zipfile.ZipFile("test.zip", "w")     # 打开压缩包，写方式


for name in glob.glob("samples/*"):        # 匹配samples目录下所有文件
    file.write(name, os.path.basename(name), zipfile.ZIP_DEFLATED)      #将文件写入到压缩包内


file.close()


file = zipfile.ZipFile("test.zip", "r")
for info in file.infolist():
    print info.filename, info.date_time, info.file_size, info.compress_size


sample.wav (1999, 8, 15, 21, 26, 46) 13260 10985
sample.jpg (1999, 9, 18, 16, 9, 44) 4762 4626
sample.au (1999, 7, 18, 20, 57, 34) 1676 1103
...
---------------------------------------------------------
使用 zipfile 模块在 ZIP 文件中储存字符串
---------------------------------------------------------
import zipfile
import glob, os, time


file = zipfile.ZipFile("test.zip", "w")          # 打开压缩包，写方式


now = time.localtime(time.time())[:6]    # 获取当前时间 (2012, 2, 3, 16, 58, 39)


for name in ("life", "of", "brian"):
    info = zipfile.ZipInfo(name)
    info.date_time = now
    info.compress_type = zipfile.ZIP_DEFLATED
    file.writestr(info, name*1000)


file.close()


# open the file again, to see what's in it


file = zipfile.ZipFile("test.zip", "r")


for info in file.infolist():
    print info.filename, info.date_time, info.file_size, info.compress_size


life (2000, 12, 1, 0, 12, 1) 4000 26
of (2000, 12, 1, 0, 12, 1) 2000 18
brian (2000, 12, 1, 0, 12, 1) 5000 31
---------------------------------------------------------
                   案例分析
---------------------------------------------------------
import zipfile
z = zipfile.ZipFile(filename, 'r') # 这里的第二个参数用r表示是读取zip文件，w是创建一个zip文件
for f in z.namelist():
    print f
---------------------------------------------------------
import zipfile
z = zipfile.ZipFile(filename, 'r')
for i in z.infolist():
    print i.file_size, i.header_offset
---------------------------------------------------------
import zipfile
z = zipfile.ZipFile(filename, 'r')
print z.read(z.namelist()[0])
---------------------------------------------------------
import zipfile, os
z = zipfile.ZipFile(filename, 'w') # 注意这里的第二个参数是w，这里的filename是压缩包的名字
#假设要把一个叫testdir中的文件全部添加到压缩包里（这里只添加一级子目录中的文件）：
if os.path.isdir(testdir):
    for d in os.listdir(testdir):
        z.write(testdir+os.sep+d)
# close() 是必须调用的！
z.close()






tar打包
在写打包代码的过程中，使用tar.add()增加文件时，会把文件本身的路径也加进去，加上arcname就能根据自己的命名规则将文件加入tar包
打包代码：
Python代码  

#!/usr/bin/env /usr/local/bin/python

  

　　 # encoding: utf-8  

　　 import tarfile  

　　 import os  

　　 import time  

　　  

　　 start = time.time()  

　　 tar=tarfile.open('/path/to/your.tar,'w')  

　　 for root,dir,files in os.walk('/path/to/dir/'):  

　　         for file in files:  

　　                 fullpath=os.path.join(root,file)  

　　                 tar.add(fullpath,arcname=file)  

　　 tar.close()  

　　 print time.time()-start  




tar解包
tar解包也可以根据不同压缩格式来解压。
Python代码  

#!/usr/bin/env /usr/local/bin/python  

• 　　 # encoding: utf-8  
  
• 　　 import tarfile  
  
• 　　 import time  
  
• 　　  
  
• 　　 start = time.time()  
  
• 　　 t = tarfile.open("/path/to/your.tar", "r:")  
  
• 　　 t.extractall(path = '/path/to/extractdir/')  
  
• 　　 t.close()  
  
• 　　 print time.time()-start  
  

上面的代码是解压所有的，也可以挨个起做不同的处理，但要如果tar包内文件过多，小心内存哦~

Python代码  

• 　　tar = tarfile.open(filename, 'r:gz')  
  
• 　　for tar_info in tar:  
  
• 　　    file = tar.extractfile(tar_info)  
  
• 　　    do_something_with(file)