增加layer—待完成[通俗易懂]

全栈程序员-用户IM • 2022年4月3日下午3:35 • 未分类

大家好，又见面了，我是你们的朋友全栈君。

 1 #! /usr/bin/env python
 2 # -*- coding:utf-8 -*-
 3 # Author Ian Ying
 4 # mail: kongqing.ying@yitu-inc.com
 5 
 6 import fileinput,sys,os,re
 7 
 8 def input_file_count(filename):
 9     '''fix last line is empty by checking last 4 lines'''
10     count_line = open("streamlayer.ev", "r")
11     #line_count = len(['' for line in count_line.readlines()])
12     line_count_arr = count_line.readlines()
13     total = len(line_count_arr)
14     for i in range(1,5):
15         print line_count_arr[-i].strip()
16         re_match = re.compile(r'^\)')
17         if re_match.search(line_count_arr[-i]):
18             line_count = total-i+1
19             print line_count_arr[-i].strip(),'in'
20             break
21         else:
22             line_count = total
23     print line_count
24     count_line.close()
25     # filename_bak = filename+'bak'
26     # filename_bak = str(filename_bak)
27     # os.rename(filename, filename_bak)
28     # old_file = open(filename_bak, 'r')
29     # with open(filename, 'w') as file_modify:
30     #     for line in old_file.readlines():
31     #         print line
32     #         file_modify.write(line)
33     return line_count
34 
35 def add_layer_streamLayer(layer,datatype,layer_num,layer_type):
36     line_count = input_file_count('streamlayer.ev')
37     print line_count
38     z = 0
39     for line in fileinput.input('streamlayer.ev',inplace=1):
40         if fileinput.lineno() == line_count and (')' in line):
41             z = 'right',fileinput.lineno()
42             print '('+ layer + ' '+datatype +' )'+ layer_num +' '+ layer_type
43             print ')'
44         elif fileinput.lineno() != line_count:
45             print line.strip()
46         else:
47             z =  'Wrong! And the total line is',fileinput.lineno()
48             print line.strip()
49             fileinput.close()
50             print " The input file structure is wrong!"
51     fileinput.close()
52     print z
53 
54 
55 if __name__ == "__main__":
56     add_layer_streamLayer('M1','drawing','3','0')