python文件讀寫-rar, zip,tar, gz壓縮，解壓縮文件

引用：https://www.jianshu.com/p/75026abc8e91

tar, gz

壓縮

import os, tarfile
#一次性打包整個根目錄。空子目錄會被打包。
#如果只打包不壓縮，將"w:gz"參數改為"w:"或"w"即可。
def make_targz(output_filename, source_dir):
  with tarfile.open(output_filename, "w:gz") as tar:
    tar.add(source_dir, arcname=os.path.basename(source_dir))

#逐個添加文件打包，未打包空子目錄。可過濾文件。
#如果只打包不壓縮，將"w:gz"參數改為"w:"或"w"即可。
def make_targz_one_by_one(output_filename, source_dir):
  tar = tarfile.open(output_filename,"w:gz")
  for root,dir,files in os.walk(source_dir):
    for file in files:
      pathfile = os.path.join(root, file)
      tar.add(pathfile)
  tar.close()

make_targz('test.tar.gz',"E:\python_sample\libs")
make_targz_one_by_one('test01.tgz',"E:\python_sample\libs")

解壓縮

import gzip
import os
def un_gz(file_name):
    """ungz zip file"""
    f_name = file_name.replace(".gz", "")
    #獲取文件的名稱，去掉
    g_file = gzip.GzipFile(file_name)
    #創建gzip對象
    open(f_name, "wb+").write(g_file.read())
    #gzip對象用read()打開后，寫入open()建立的文件里。
    g_file.close() #關閉gzip對象

   
import tarfile
def un_tar(file_name):
    #untar zip file
    tar = tarfile.open(file_name)
    names = tar.getnames()
    if os.path.isdir(file_name + "_files"):
        pass
    else:
        os.mkdir(file_name + "_files")
    #由於解壓后是許多文件，預先建立同名文件夾
    for name in names:
        tar.extract(name, file_name + "_files/")
    tar.close()
un_gz("test.tar.gz")
un_tar("test.tar")

zip

壓縮

import os, zipfile
#打包目錄為zip文件（未壓縮）
def make_zip(source_dir, output_filename):
    zipf = zipfile.ZipFile(output_filename, 'w')
    pre_len = len(os.path.dirname(source_dir))
    for parent, dirnames, filenames in os.walk(source_dir):
        for filename in filenames:
            print(filename)
            pathfile = os.path.join(parent, filename)
            arcname = pathfile[pre_len:].strip(os.path.sep)   #相對路徑
            zipf.write(pathfile, arcname)
        print()
    zipf.close()

make_zip(r"E:\python_sample\libs\test_tar_files\libs","test.zip")

解壓縮

import os,zipfile
def un_zip(file_name):
    """unzip zip file"""
    zip_file = zipfile.ZipFile(file_name)
    if os.path.isdir(file_name + "_files"):
        pass
    else:
        os.mkdir(file_name + "_files")
    for names in zip_file.namelist():
        zip_file.extract(names,file_name + "_files/")
    zip_file.close()
un_zip("test.zip")

rar

解壓縮

#rarfile不支持創建rar壓縮卷,請用zip/7z
import rarfile
def unrar(rar_file, dir_name):
# rarfile需要unrar支持,
# linux下pip install unrar, windows下在winrar文件夾找到unrar,加到path里
rarobj = rarfile.RarFile(rar_file.decode('utf-8'))
rarobj.extractall(dir_name.decode('utf-8'))