有時候寫了一堆文件，從哪里抄來的，或者字節寫的，沒有按照linux kernel的代碼標準，這時候手動去改，非常繁瑣
.
寫一個python腳本, 自動將源代碼文件(如.c文件)格式化/規范化, 如刪除多余的空行空格, 按照linux kernel標準格式化.
當前的使用方法 : python3 format.py 跟上文件名, 后續可擴展對目錄下所有文件操作.

import sys import osconditional_str = ["if","while","for" ]def table_space(tag_count):total_str=""if tag_count == 0:return total_strfor num in range(0,tag_count):total_str += '\t'return total_strdef is_space_line(line):if '' == line.strip():return Trueelse:return Falsedef delete_redundant_space(flag_dict):flag_dict['line'] = flag_dict['line'].strip() + '\n'flag_dict['line_need'] = Truedef insert_space_char(flag_dict):line = flag_dict['line']for con_str in conditional_str:temp_val = con_str + '('if temp_val in line:nPos=flag_dict['line'].index(temp_val)list_line = list(flag_dict['line'])list_line.insert(nPos + len(temp_val) - 1,' ')flag_dict['line'] = "".join(list_line)def calculate_table_space(flag_dict):line = flag_dict['line']while True:if '#' == line[0]:flag_dict['last_tag_cnt'] = 0breakif '\n' == line:# print("=======kong hang=========")breakif '{' in line and '}' in line:break;if '{' in line:flag_dict['tag_cnt'] += 1if '}' in line:flag_dict['tag_cnt'] -= 1flag_dict['last_tag_cnt'] -= 1breakdef add_table_space_to_line(flag_dict):flag_dict['line'] = table_space(flag_dict['last_tag_cnt'] ) + flag_dict['line']def judge_conditional_statement(flag_dict,fp2):line = flag_dict['line']flag_dict['line'] = flag_dict['line']if flag_dict['undoing_if'] == True:if is_space_line(line) == True:flag_dict['line_need'] = Falsereturnflag_dict['undoing_if'] = Falseif '\t{' in line:flag_dict['line_need'] = Falseelse:fp2.seek(fp2.tell() - 3)fp2.write('\n')flag_dict['line'] = '\t' + flag_dict['line']returnfor con_str in conditional_str:temp_val1 = '\t' + con_str + ' 'temp_val2 = '\t' + con_str + '('if (temp_val1 in line or temp_val2 in line) and ')' in line and '{' not in line:flag_dict['line_need'] = Falseflag_dict['last_line_need'] = Trueflag_dict['last_line'] = flag_dict['line'].rstrip() + " {\n"flag_dict['undoing_if'] = Truereturndef format_a_file(srcfile,dstfile):flag_dict = {'line_need' : False,'line' : "",'last_line_need' : False,'last_line' : "",'tag_cnt' : 0,'last_tag_cnt' : 0,'undoing_if' : False,}with open(srcfile, 'r') as fp1, open(dstfile, 'w') as fp2:lines = fp1.readlines()for line in lines:flag_dict['line'] = lineflag_dict['last_tag_cnt'] = flag_dict['tag_cnt']delete_redundant_space(flag_dict)calculate_table_space(flag_dict)add_table_space_to_line(flag_dict)insert_space_char(flag_dict)if flag_dict['last_line_need'] == True:fp2.write(flag_dict['last_line'])flag_dict['last_line_need'] = Falsejudge_conditional_statement(flag_dict,fp2)if flag_dict['line_need'] == True:fp2.write(flag_dict['line'])def format_all_files(srcfile):print("1111111111")if __name__ == '__main__':print(str(sys.argv[0]) + " enter")print(len(sys.argv))print(os.getcwd())# srcfile = sys.argv[1]# dstfile = sys.argv[2]srcfile = "FtModule.cpp"dstfile = '1.txt'if True:format_all_files(srcfile)exitprint(("Format %s to %s") % (srcfile,dstfile))curdir = os.getcwd()print(curdir)#print(table_space(3) + 'wwww')format_a_file(srcfile,dstfile)#os.rename(srcfile, 'FtModule.cpp')

總結

以上是生活随笔為你收集整理的[工具]-脚本自动化工具:按照linux kernel标准格式化输出文件(format_file)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。