【自动化测试】Python—zipfile库文件压缩封装

【自动化测试】Python—zipfile库文件压缩封装,第1张

当自动化测试执行完测试用例后,生成的测试用例报告(Html、Allure等)一般会放在某个目录中,如果想把测试报告通过邮件的方式自动发送给其他人,这时候前提是需要把该目录下的测试报告们打包压缩,然后再邮件发送。

一、“.zip”压缩文件类型

在我们使用zipfile时,zipfile处理的压缩文件后缀名可是“.zip”,“.zip”对应的是winzip软件,在我们使用zipfile进行压缩 *** 作后,也需要相应的软件进行检验查阅 *** 作,下边介绍winzip软件。

zip格式比较常见支技泛围广,windows *** 作系统不装任何第三方软件也可以打开zip格式

winzip是付费软件,使用界面如下:

个人用的比较多的是7zip,其实winzip之前几乎没有接触过,在了解python压缩库的时候才把winzip下载了,因为zipfile库需要用到winzip,其实两个压缩软件使用体验差不太多,winzip可能更加专业化,不过需要付费。

下载

winzip下载地址:WinZip Windows 版 - 压缩文件、解压缩文件

二、zipfile日常使用

简单来说zipfile模块用来做zip格式编码的压缩和解压缩的,ZipFile是主要的类,用来创建和读取zip文件。

1、ZipFile 对象

class zipfile.ZipFile(file, mode='r', compression=ZIP_STORED)

zipfile要进行相关 *** 作,首先实例化一个 ZipFile 对象,我选取了最为常用的三个参数,“file”是必选参数,指字符串格式压缩包名称(也可指定路径);“mode”是可选参数,表示打开模式/类似文件 *** 作,有r、w、a三种模式,分别表示读、写、添加 *** 作;

“compression”是可选参数,指在写入压缩文件时使用的 ZIP 压缩方法👇,目的是模仿各压缩方法的数字常数,不过最终读取或创建的都是zip压缩算法下的文件。(本人试验过,不论哪种compression,创建新的压缩文件依然是“.zip” *** 作)

zipfile.py模块:

 

2、zipfile读取

比如要读取一个zip压缩包,这里假设filename是该压缩包的路径,z.namelist() 会返回压缩包内所有文件名的列表,for循环打印出列表的值,也就是文件名。

import zipfile  
zipf = zipfile.ZipFile(filename, 'r') 
# 这里的第二个参数用r表示是读取zip文件,w是创建一个zip文件  
for f in zipf.namelist(): 
    print(f)

z.infolist()返回的是压缩包内所有文件的信息(单目录),是一个ZipInfo的列表,里边包含了多个ZipInfo对象,每个对象包含了压缩文件的信息,filename, file_size, header_offset, 分别为文件名,文件大小,文件数据在压缩包中的偏移。

import zipfile  
zipf = zipfile.ZipFile(filename, 'r')  
for i in zipf.infolist():  
    print(i.filename,i.file_size)#打印文件名和大小

读:从压缩包里解压出一个文件的方法是使用ZipFile的read方法,这样读取出z.namelist()中的第一个文件,并且输出到屏幕,也支持pwd参数(压缩文件密码)。

import zipfile  
zipf = zipfile.ZipFile(filename, 'r')  
print zipf.read(zipf.namelist()[0],pwd=None) 

提取:zipf.extract()和zipf.extractall()分别是提取压缩包里的某个文件或所以文件,path是指定的解压出的目录路径,也支持pwd参数(压缩文件密码)。

zipf =zipfile.ZipFile(r'C:\Users\xiaowo\Desktop\xx1.zip', 'r')
# 这里的第二个参数用r表示是读取zip文件,w是创建一个zip文件
zipf.extract(zipf.namelist()[0], path=None, pwd=None) #提取压缩包里的某个压缩文件
zipf.extractall(path=None, pwd=None) #提取压缩包里的所以压缩文件
3、zipfile写入

写入需要用到ZipFile.write(filename, arcname=None, compress_type=None)函数,filename是必选参数,指文件名/路径,arcname指的是写入的压缩包里的文件的名称,默认情况下与filename一致。

import zipfile, os  
z = zipfile.ZipFile(filename, 'w') 
# 注意这里的第二个参数是w,这里的filename是压缩包的名字
if os.path.isdir(testdir):  #判断testdir路径是否是目录
     for d in os.listdir(testdir):  #返回指定的文件夹包含的文件或文件夹的名字的列表。
         z.write(testdir+os.sep+d)  
     z.close() 

三、zipfile文件压缩封装

1.导入zipfile、re库,定义make_zip函数,也就是zip打包压缩

2.实例化一个ZipFile 对象,注意mode是’w‘,表示写入

3.遍历指定文件夹里的文件,对其依次进行写入 *** 作,循环结束后关闭ZipFile 对象

import zipfile,re
def make_zip(localPath,pname): #localPath:待压缩的文件包路径,pname:写入新的压缩文件名(创建在当前py文件目录下),也可指定具体路径
    '''打包zip'''
    if re.search('.zip',pname): #re.search整个pname字符串,返回一个成功的匹配   #因为新的压缩文件带.zip才能正常使用,不然需要人为添加
        pass            #提升函数灵活性,调用函数时,pname携带或不携带“.zip”均可
    else:
        pname = pname +'.zip'
    zipf = zipfile.ZipFile(pname,'w',zipfile.ZIP_DEFLATED)  #实例化zipfile对象,表示一个zip文件,pname时文件名(创建在当前py文件目录下)
    if os.path.isdir(localPath):  # 判断testdir路径是否是目录
        for d in os.listdir(localPath):  # 返回指定的文件夹包含的文件或文件夹的名字的列表。
            z.write(localPath + os.sep + d)
        z.close()
    return pname
总结

大家如果有疑问都可以评论提出,有不足之处请大家批评指正,希望能多结识这方面的朋友,共同学习、共同进步。

欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/796380.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-05-06
下一篇 2022-05-06

发表评论

登录后才能评论

评论列表(0条)

保存