字符串替换批量修改脚本

针对客户需求，会遇到测试用例中Counter名称改变的问题，此脚本主要解决批量Case文件修改Counter名称的问题，源码如下：

 1 #!/usr/bin/python
 2 # -*- coding:utf-8 -*-
 3 
 4 import sys,os
 5 import xlrd
 6 
 7 # 枚举当前目录下的所有.mp文件,并返回list
 8 def get_filename(): 
 9 #begin  
10     fileDir = "." + os.sep
11     L = []
12     for root, dirs, files in os.walk(fileDir):  
13         for file in files:
14             if os.path.splitext(file)[1] == '.mp':
15                 L.append(os.path.join(file))
16     #print L
17     return L  
18 #end                   
19                    
20 #通过读取Excel文件的方式获取旧的counter名称和新的counter名称，存为list
21 #并将所有get到的counter名称按照长度从大到小进行排序，防止短counter名称先替换
22 #实际执行中发现长counter名称包含短counter名称，而如果短counter先与长counter替换，则导致长counter无法替换
23 def get_countername():
24     fname = "counterfile\CounterNameAll.xls"
25     bk = xlrd.open_workbook(fname)
26     shxrange = range(bk.nsheets)
27     try:
28       sh = bk.sheet_by_name("Sheet1")
29     except:
30       print "no sheet in %s named Sheet1" % fname
31     #获取行数
32     nrows = sh.nrows
33     #获取列数
34     ncols = sh.ncols
35     print "nrows %d, ncols %d" % (nrows,ncols)
36 
37     counter_list = []
38     row_list = []
39     #获取各行数据
40     for i in range(0,nrows):
41       row_data = sh.row_values(i)
42       row_list.append(row_data)
43     #对get到的二维数组进行排序
44     row_list = sorted(row_list, key=lambda row:len(row[1]), reverse=True)  
45     for i in range(0,nrows):
46         for j in range(0,ncols):
47             counter_list.append(row_list[i][j].encode('utf-8'))
48     new_list = []
49     for count in counter_list:
50       new_list.append(count.replace('\xc2\xa0', ' '))
51     for i in range(0,len(new_list),2):
52       new_list[i] = ' '*(len(new_list[i+1])-len(new_list[i])) + new_list[i]
53     print new_list
54     return new_list  
55 
56 
57 
58 #把指定文件的内容替换的函数,多个环境参数并行替换，这样就可以一个脚本，多个通用。
59 def repip_func(file_path, counter_list):
60 
61   f = open(file_path,'r+')
62   all_the_lines = f.readlines()
63   f.seek(0)
64   f.truncate()
65   for line in all_the_lines:
66     i = 0
67     while i < len(counter_list):
68         line = line.replace(counter_list[i+1],counter_list[i])
69         i = i + 2
70     f.write(line)
71   f.close()
72 
73   #print "%s have replace ip %s" % (file_path, counter_list)
74 
75 def main():
76   repip_file_list = get_filename()
77   counter_list = get_countername()
78   for file_path in repip_file_list:
79     repip_func(file_path, counter_list)
80   #print counter_list
81 
82 if __name__ == "__main__":
83   main()