python 代码格式规范脚本_python脚本处理代码注释规范化

python脚本 实现注释的规范化处理：

示例：

1.

源代码：

int speed = 4;//设置速度为4

处理后：

//设置速度为4

int speed = 4;

2.

源代码：

else{//否则

处理后：

//否则

else{

3.

源代码：

}else{//否则

处理后：

//否则

}else{

4.

源代码：

if(condition == true){//如果为真

处理后：

//如果为真

if(condition == true){

5.

源代码：

json处理：(js)

var testJson = {

name:”lmn”,//名字

age:”25”//年龄

}

处理后：

不进行处理。

说多无益，直接上代码：

#encoding:utf-8

'''

可能性如下：

1.单独一行注释：

2.有代码有注释

3 }else{//

4.多行注释

/**

* //获取所有配件重量总和

* @param index 将要装备的第几个配件槽

* @param unitId 将要装备配件ID

*/

5.

1: "tankmuzzle",//炮管 11

满足要求：

所有的注释都单独一行，除了JSON的注释

保持与下一行同样缩进

'''

import os

#文件列表

fileList = []

writeFileList = []

#代码注释格式声明

codeAnnotationDec = '//'

# str:将要处理的code saveList:将要存入的列表

def dealCodeWithSpace(lines,willSaveList):

# lines=lines.strip('\n')

#计算代码首部缩进空格长度 去掉尾部空格长度 - 去掉首尾空格长度

headSpaceLen = len(lines.rstrip())-len(lines.strip())

#先添加注释(注释代码+首部缩进)

# willSaveList.append(" "*headSpaceLen+lines[lines.find(codeAnnotationDec):len(lines)])

tempAnno = lines[lines.find(codeAnnotationDec): ].rstrip()

if tempAnno.find('\n')>-1:

willSaveList.append(" "*headSpaceLen+tempAnno)

else:

willSaveList.append(" "*headSpaceLen+tempAnno+'\n')

willSaveList.append('\n')

if len(willSaveList)>=2:

if willSaveList[len(willSaveList) -1] == '\n' and willSaveList[len(willSaveList) -2].find(codeAnnotationDec)>-1:

willSaveList.pop()

#再添加代码：

tempCode = lines[0:lines.find(codeAnnotationDec)]

if tempCode.find('\n')>-1:

willSaveList.append(tempCode)

else:

willSaveList.append(tempCode)

willSaveList.append('\n')

# willSaveList.append(lines[0:lines.find(codeAnnotationDec)].rstrip()+'\n')

def doDealCode(readFilePath,writeFile):

#如果传递进来的是文件，就直接加入处理文件列表

if os.path.isfile(readFilePath):

fileList.append(dir.decode("utf-8"))

writeFileList.append(dir.decode("utf-8"))

#把读取目录的所有文件加入文件列表

elif os.path.isdir(readFilePath):

for singleFile in os.listdir(readFilePath):

fileList.append(os.path.join(readFilePath,singleFile))

writeFileList.append(os.path.join(writeFile,singleFile))

if 1:

fileIndex = 0

for readSingleFile in fileList:

#处理后的代码行，每行都会插入列表中

willSaveList =[]

fileIndex = fileList.index(readSingleFile)

#初步处理文件行，依次加入一个列表中

with open(readSingleFile,'r') as fileLine:

while True:

#整行读取数据

lines = fileLine.readline()

if not lines:

break

#如果该行不为空

else:

# 该行有 单行注释

if lines.find(codeAnnotationDec)>-1:

#该行为存粹一行注释

if lines.strip().find(codeAnnotationDec) == 0:

willSaveList.append(lines)

#该行既有代码，也有注释

else:

#处理json文件，默认注释和代码放在一行：

if lines.find(":")>-1:

willSaveList.append(lines)

#处理 }else{ 情况 #存在三种情况：1： else{//dfsdfdsfsdf 2:}else{//dsfdfdsf 3.if(j == index){//计算的重量,用将要替换的配件的重量

elif len(lines[0:lines.find(codeAnnotationDec)].rstrip())>0 and lines[0:lines.find(codeAnnotationDec)].rstrip()[-1] == '{':

#第一种情况处理：

if lines.strip().find("else") == 0:

dealCodeWithSpace(lines,willSaveList)

#第二种 和第三种 情况处理

else:

#第三种情况：

if lines.strip().find('if') == 0:

dealCodeWithSpace(lines,willSaveList)

#第二种情况

else:

#计算代码首部缩进空格长度 去掉尾部空格长度 - 去掉首尾空格长度

headSpaceLen = len(lines.rstrip())-len(lines.strip())

# 1.在 else 前面插入一行：

willSaveList.append(lines[0:lines.find("else")]+'\n')

# 求出将要处理的字符串 else{ 缩进要和上一行 { 保持一致

willDealCode = headSpaceLen*' '+lines[lines.find("else"): ]

# 2.按照 code;//dsfdsfsdf 的格式进行处理

dealCodeWithSpace(willDealCode,willSaveList)

#不含特殊情况的注释情况：例如：code;//annotation

else:

dealCodeWithSpace(lines,willSaveList)

#该行没有注释(一行代码或者换行)

else:

# willSaveList[len(willSaveList)-1].scrip().find('\n') == 0:

willSaveList.append(lines)

pass

#写入同名文件到另一个目录下

with open(writeFileList[fileIndex],'w') as tempWrite:

print("------ WRITE SUC------",writeFileList[fileIndex]);

print(writeFileList[fileIndex])

for line in willSaveList:

tempWrite.write(line)

# print("******* SUC *******");

#读取文件的路径

readFilePath = "./readFile"

#写入文件的路径

writeFile = "./writeFile"

#开始处理文件

doDealCode(readFilePath,writeFile)