python开发_tarfile_文档归档压缩|解压缩

'''
    python中的tarfile模块实现文档的归档压缩和解压缩

    功能：
        把工作空间下面的所有文件，打包生成一个tar文件
        同时提供一个方法把该tar文件中的一些文件解压缩到
        指定的目录中
'''

运行效果：

如果在你计算机的C盘里面没有test目录，系统会自动创建该目录：c:\\test

如果在你计算机的c:\\test目录下面没有temp目录，系统会自动创建该目录：c:\\test\\temp

控制台输出：

Python 3.3.2 (v3.3.2:d047928ae3f6, May 16 2013, 00:03:43) [MSC v.1600 32 bit (Intel)] on win32
Type "copyright", "credits" or "license()" for more information.
>>> ================================ RESTART ================================
>>>
创建目录:[c:\test]
创建目录:[c:\test\temp]
获取工作空间下的所有文件...
打开文件:[c:\test\hongten.tar]
压缩文件:[fibo.py]
压缩文件:[main.py]
压缩文件:[pythonindex.html]
压缩文件:[test_arrange.py]
压缩文件:[test_array.py]
压缩文件:[test_blogs.py]
压缩文件:[test_calendar.py]
压缩文件:[test_calendar_html.py]
压缩文件:[test_class.py]
压缩文件:[test_collections]
压缩文件:[test_collections.py]
压缩文件:[test_copy.py]
压缩文件:[test_count.py]
压缩文件:[test_email.py]
压缩文件:[test_email_send.py]
压缩文件:[test_filecmp.py]
压缩文件:[test_for.py]
压缩文件:[test_function.py]
压缩文件:[test_glob.py]
压缩文件:[test_gzip.py]
压缩文件:[test_linecache.py]
压缩文件:[test_list.py]
压缩文件:[test_modules.py]
压缩文件:[test_more_mudels.py]
压缩文件:[test_mysql.py]
压缩文件:[test_n.py]
压缩文件:[test_os.py]
压缩文件:[test_pprint.py]
压缩文件:[test_qq.py]
压缩文件:[test_shutil.py]
压缩文件:[test_stat.py]
压缩文件:[test_tar.py]
压缩文件:[test_tempfile.py]
压缩文件:[test_tkinter.py]
压缩文件:[test_truth_value.py]
压缩文件:[test_wifi.py]
压缩文件:[test_windows.py]
压缩文件:[__pycache__]
关闭文件[c:\test\hongten.tar]
##################################################
打开文件:[c:\test\hongten.tar]
获取到所有文件名称:['fibo.py', 'main.py', 'pythonindex.html', 'test_arrange.py', 'test_array.py', 'test_blogs.py', 'test_calendar.py', 'test_calendar_html.py', 'test_class.py', 'test_collections', 'test_collections.py', 'test_copy.py', 'test_count.py', 'test_email.py', 'test_email_send.py', 'test_filecmp.py', 'test_for.py', 'test_function.py', 'test_glob.py', 'test_gzip.py', 'test_linecache.py', 'test_list.py', 'test_modules.py', 'test_more_mudels.py', 'test_mysql.py', 'test_n.py', 'test_os.py', 'test_pprint.py', 'test_qq.py', 'test_shutil.py', 'test_stat.py', 'test_tar.py', 'test_tempfile.py', 'test_tkinter.py', 'test_truth_value.py', 'test_wifi.py', 'test_windows.py', '__pycache__', '__pycache__/fibo.cpython-33.pyc']
提取文件：[pythonindex.html]
>>> 

=================================================

代码部分：

=================================================

 #python tar

 #Author : Hongten
 #MailTo : hongtenzone@foxmail.com
 #QQ     : 648719819
 #Blog   : http://www.cnblogs.com/hongten
 #Create : 2013-08-19
 #Version: 1.0

 import os
 import tarfile
 '''
     python中的tarfile模块实现文档的归档压缩和解压缩

     功能：
         把工作空间下面的所有文件，打包生成一个tar文件
         同时提供一个方法把该tar文件中的一些文件解压缩到
         指定的目录中
 '''
 #global var
 SHOW_LOG = True
 #tar文件存放位置
 TAR_PATH = ''
 #取出文件存放目录
 EXT_PATH = ''

 def write_tar_file(path, content):
     '''打开指定path的tar格式的文件，如果该文件不存在
     系统会自动创建该文件，如果该文件以及存在，则打开文件
     打开文件后，向文件中添加文件(这个功能类似于把几个文件
     打包成tar包文件)'''
     with tarfile.open(path, 'w') as tar:
         if SHOW_LOG:
             print('打开文件:[{}]'.format(path))
         for n in content:
             if SHOW_LOG:
                 print('压缩文件:[{}]'.format(n))
             tar.add(n)
         if SHOW_LOG:
             print('关闭文件[{}]'.format(path))
         tar.close()

 def get_workspace_files():
     '''获取工作空间下面的所有文件，然后以列表的形式返回'''
     if  SHOW_LOG:
         print('获取工作空间下的所有文件...')
     return os.listdir('./')

 def extract_files(tar_path, ext_path, ext_name):
     '''解压tar文件中的部分文件到指定目录中'''
     with tarfile.open(tar_path) as tar:
         if SHOW_LOG:
             print('打开文件:[{}]'.format(tar_path))
         names = tar.getnames()
         if SHOW_LOG:
             print('获取到所有文件名称:{}'.format(names))
         for name in names:
             if name.split('.')[-1] == ext_name:
                 if SHOW_LOG:
                     print('提取文件：[{}]'.format(name))
                 tar.extract(name, path = ext_path)

 def mkdir(path):
     '''创建不存在的目录'''
     if os.path.exists(path):
         if SHOW_LOG:
             print('存在目录:[{}]'.format(path))
     else:
         if SHOW_LOG:
             print('创建目录:[{}]'.format(path))
         os.mkdir(path)

 def init():
     global SHOW_LOG
     SHOW_LOG = True
     #tar文件存放位置
     global TAR_PATH
     TAR_PATH = 'c:\\test\\hongten.tar'
     #取出文件存放目录
     global EXT_PATH
     EXT_PATH = 'c:\\test\\temp'
     #创建目录，如果目录不存在
     path = os.path.split(TAR_PATH)[0]
     mkdir(path)
     mkdir(EXT_PATH)

 def main():
     init()
     content = get_workspace_files()
     #打包文件
     write_tar_file(TAR_PATH, content)
     print('#' * 50)
     #提取文件
     extract_files(TAR_PATH, EXT_PATH, 'html')

 if __name__ == '__main__':
     main()