1 import os,sys
2 import csv
3
4 file_folder = sys.argv[1]
5 # file_folder = r"/Users/vito/Desktop/0120 "
6 final_info={}
7 txt_path_list=[]
8
9 # 如需要处理更多log信息，在下面三行最后面添加; 如果读不到数据 可以检查关键字是否一致
10 fileHead = ('distance', 'center_4edge_mean_MTF50_Y','left_diff_MTF50_Y')
11 request_testItem = ('center_4edge_mean_MTF50_Y','left_diff_MTF50_Y')
12 distance_list = ["1750", "1755", "1760", "1765", "1770", "1775", "1780", "1785", "1790", "1795", "1800", "1805", "1810", "1815", "1820", "1825", "1830", "1835", "1840", "1845", "1850"]
13
14 def handle_txt(txt_path_list,final_info):
15 for path in txt_path_list:
16 with open(path,'r') as f:
17 content = f.readlines()
18 log_info = {}
19 for line in content:
20 handled_line_info = line.strip().split(':=') # name:=值
21 # print(handled_line_info)
22 if handled_line_info[0] in request_testItem:
23 log_info[handled_line_info[0]]=handled_line_info[1] # {name:值}
24 for distance in distance_list:
25 if distance in path:
26 final_info[distance]=log_info # 字典格式为：distance:{name:值，name:值，name:值}
27 return final_info
28
29 def update_csv_log(path,final_info):
30 with open(path,'w') as f:
31 wr = csv.writer(f, dialect='excel')
32 wr.writerow(fileHead)
33 for distance,txt_info in final_info.items():
34 data_value = []
35 for item in request_testItem:
36 item_value = final_info[distance].get(item)
37 data_value.append(item_value)
38 new_row = (distance,*tuple(data_value))
39 wr.writerow(new_row)
40
41 def get_txt_path(filefolder,txt_path_list):
42 tmp_path_List = os.listdir(filefolder)
43 for File in tmp_path_List:
44 subPath = os.path.join(filefolder,File)
45 if os.path.isdir(subPath):
46 get_txt_path(subPath,txt_path_list)
47 elif subPath.endswith('txt'):
48 txt_path_list.append(subPath)
49 txt_path_list.sort()
50 # print(txt_path_list)
51 return txt_path_list
52
53
54 txt_path_list = get_txt_path(file_folder,txt_path_list)
55 final_info = handle_txt(txt_path_list,final_info)
56 update_csv_log(file_folder+'/WD_summary.csv',final_info)
