在工作过程中,需要将在某一个文件下面的所有ANVLTestReport.csv进行汇总,首先先要获取到这个文件所有的路径,可以通过os模块下面的os.walk函数实现该功能
os.walk(目录)返回的是一个生成器,我们需要不断遍历它,才可以获取到它所有的内容,os.walk默认是从最顶级的目录开始遍历
例如:
- import os
- a=os.walk(r‘C:\IxANVL\Xanvl-data\Results\IPv6 Test Suites.resDir\ICMPv6.resDir\icmpv6.res’)
- print(next(a))
运行之后返回结果生成器第一个值:
('C:\\IxANVL\\Xanvl-data\\Results\\IPv6 Test Suites.resDir\\ICMPv6.resDir\\icmpv6.res', ['Run0001.res', 'Run0002.res', 'Run0003.res', 'Run0004.res', 'Run0005.res', 'Run0006.res'], ['.count', 'Run0006.res.rar'])
返回一个三元组,元组里面第一个是目录名,第二个是一个列表,表示文件夹名,第三个也是列表,表示目录下面的文件名
如果想查看该目录下面是否存在指定文件,则只用确认三元组中第三个列表是否有该文件即可;
实现如下:
- import os
- def file_path(dir, file_name):
- file_path = []
- for i in os.walk(dir):
- if file_name in i[2]:
- file_dirpath = i[0]
- path = os.path.join(file_dirpath, file_name)
- file_path.append(path)
- return file_path
- dir=‘C:\\IxANVL\\Xanvl-data\\Results\\IPv6 Test Suites.resDir\\ICMPv6.resDir\\icmpv6.res’
- file_name=‘ANVLTestReport.csv’
- print(file_path(dir,file_name))
执行之后结果为:
[‘C:\\IxANVL\\Xanvl-data\\Results\\IPv6 Test Suites.resDir\\ICMPv6.resDir\\icmpv6.res\\Run0003.res\\ANVLTestReport.csv’, ‘C:\\IxANVL\\Xanvl-data\\Results\\IPv6 Test Suites.resDir\\ICMPv6.resDir\\icmpv6.res\\Run0006.res\\ANVLTestReport.csv’]