python3，为某一路径下所以文件内容特定字段前添加特定内容，\n前添加特定字符

心已赠人 2022-11-20 10:59 50阅读 0赞

# encoding: utf-8
    
    
    import os
    import re
    
    def filecontentdeal(filecontent,addi):
    	if filecontent.find('\\n') != -1:
    		contentsplit = filecontent.split('\\n')
    		endlistcontent = contentsplit.pop()
    		retcontent = ''
    		for i in contentsplit:
    			if (len(i) >= 1) and (i[-1] == '\''):
    				addstr = i + "\\n"
    			else:
    				print("print addi is " + str(addi))
    				#[cyf]%d[cyf]这是添加的标签，方便查找，[cyf]可以更改
    				addstr = i + "[cyf]" + str(addi) + "[cyf]\\n"
    				addi = addi + 1
    			retcontent = retcontent + addstr
    		return (retcontent + endlistcontent),addi
    	else:
    		return filecontent,addi
    
    	
    def filecontentdealold(filecontent):
    	if filecontent.find('[cyf]') != -1:
    		strinfo = re.compile(r'\[cyf\][0-9]{0,9}\[cyf\]')
    		retcontent = strinfo.sub('',filecontent)
    		return retcontent
    	else:
    		return filecontent
    	
    	
    def fileoper(rootpath):
    	addi = 0
    	for root,dirs,files in os.walk(rootpath):
    		for file in files:
    			filepath = os.path.join(root,file)
    			#排除特殊路径的文件不加入修改防止误修改配置文件等
    			if filepath.find('config') != -1 or filepath.find('build')!= -1:
    				continue
    			
    			#.c file and .h file
    			if filepath.endswith('.c') or filepath.endswith('.h'):
    				print(filepath)
    				fp = open(filepath,'r')
    				try:
    					filecontent = fp.read()
    				except:
    					fp.close()
    					print('read file' + filepath + 'failed-----------------')
    					continue
    				fp.close()
    				
    				#给\n前面打上标签，不包括字符'\n'
    				filecontent,addi = filecontentdeal(filecontent,addi)
    				#还原之前打过的标签删掉,需要还原的可以打开此注释
    				#filecontent = filecontentdealold(filecontent)
    				fp = open(filepath,'w')
    				try:
    					fp.write(filecontent)
    				except:
    					fp.close()
    					print('write file' + filepath + 'failed-----------------')
    					continue
    				fp.close()
    	
    
    if __name__ == "__main__":
        fileoper(r'./')