1.概述
- 在平時自動化測試工作中,經(jīng)常會用python對一些文件進(jìn)行讀寫操作。其中使用最多的文件格式,就是txt, log, json, csv, xml, zip, tar, gz, rar, excel,這十種文件格式。
- rar, zip,tar, gz這四種格式,需要使用python的第三方庫進(jìn)行操作。
2.簡介
- gz: 即gzip,通常只能壓縮一個文件。與tar結(jié)合起來就可以實現(xiàn)先打包,再壓縮。
- tar: linux系統(tǒng)下的打包工具,只打包,不壓縮
- tgz:即tar.gz。先用tar打包,然后再用gz壓縮得到的文件
- zip: 不同于gzip,雖然使用相似的算法,可以打包壓縮多個文件,不過分別壓縮文件,壓縮率低于tar。
- rar:打包壓縮文件,最初用于DOS,基于window操作系統(tǒng)。壓縮率比zip高,但速度慢,隨機訪問的速度也慢
3.tar, gz
import os, tarfile
#一次性打包整個根目錄。空子目錄會被打包。
#如果只打包不壓縮,將"w:gz"參數(shù)改為"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"參數(shù)改為"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)
#創(chuàng)建gzip對象
open(f_name, "wb+").write(g_file.read())
#gzip對象用read()打開后,寫入open()建立的文件里。
g_file.close() #關(guān)閉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")
#由于解壓后是許多文件,預(yù)先建立同名文件夾
for name in names:
tar.extract(name, file_name + "_files/")
tar.close()
un_gz("test.tar.gz")
un_tar("test.tar")
4.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")
5.rar
#rarfile不支持創(chuàng)建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'))