删除某文件夹(可包含其他文件夹)中Java文件的注释(Python)

#coding:utf-8

import os, sys

def help():
print """命令行格式: DelComment srcpath [dstpath]
srcpath为源目录或文件的路径；
detpath必须是已经存在的，它是存放转换后的目录的路径，默认为当前工作目录。
"""

def stripSlash(str):
return str.rstrip('\\').rstrip('/')

def copyRemoveAllComment(srcpath, dstpath):
#定义扫描的状态
STATE_SCAN = 0
STATE_IS_COMMENT_START = 1
STATE_IN_BLOCK_COMMENT = 2
STATE_IN_LINE_COMMENT = 3
STATE_IS_COMMENT_END = 4

#在dstpath中打开文件
srcfile = open(srcpath, 'rb')
dstfile = open(dstpath, 'wb')

#过滤注释
readState = STATE_SCAN
text = []
while True:
ch = srcfile.read(1)

if ch == '':
break;
if readState == STATE_SCAN:
if ch == '/':
readState = STATE_IS_COMMENT_START
else:
text.append(ch)
elif readState == STATE_IS_COMMENT_START:
if ch == '/':
readState = STATE_IN_LINE_COMMENT
elif ch == '*':
readState = STATE_IN_BLOCK_COMMENT
else:
readState = STATE_SCAN
text.append('/'+ch)
elif readState == STATE_IN_LINE_COMMENT:
if ch == '\n':
readState = STATE_SCAN
elif readState == STATE_IN_BLOCK_COMMENT:
if ch == '*':
readState = STATE_IS_COMMENT_END
elif readState == STATE_IS_COMMENT_END:
if ch == '/':
readState = STATE_SCAN
ch = srcfile.read(1)
while ch == '\r' or ch == '\n':
ch = srcfile.read(1)
srcfile.seek(-1, 1)
else:
readState = STATE_IN_BLOCK_COMMENT
srcfile.seek(-1, 1)

#写文件
dstfile.writelines(text)
dstfile.flush()
#关闭文件
srcfile.close()
dstfile.close()

def convert(srcpath, dstpath):
if os.path.isdir(srcpath):
dirfiles = os.listdir(srcpath)
for fn in dirfiles:
srcfp = os.path.join(srcpath, fn)
if os.path.isfile(srcfp):
convert(srcfp, dstpath)
else:
dstfp = os.path.join(dstpath, fn)
if not os.path.exists(dstfp):
os.mkdir(dstfp)
convert(srcfp, dstfp)
else:
if srcpath.endswith('.java'):
srcfn = srcpath
dstfn = os.path.join(dstpath, os.path.basename(srcpath))
copyRemoveAllComment(srcfn, dstfn)

if __name__ == '__main__':
argvlen = len(sys.argv)
#至少一个参数
if argvlen > 2+1 or argvlen <= 2-1:
print '参数错误！:-(\n'
help()
sys.exit(1)
elif not os.path.isfile(sys.argv[1]) and not os.path.isdir(sys.argv[1]) or argvlen == 1 + 2 and os.path.isfile(sys.argv[1]):
print '参数错误！:-(\n'
help()
sys.exit(2)
#获取源路径
srcpath = stripSlash(sys.argv[1])
print 'srcpath: ' + srcpath

#获取目标路径,
dstpath = ""
if argvlen == 1 + 2:
dstpath = stripSlash(sys.argv[2])
else:
dstpath = os.getcwd() + '\\delcomment'
if not os.path.exists(dstpath):
os.mkdir(dstpath)

#如果源路径是文件，则直接放在目标路径下
if os.path.isdir(srcpath):
dstpath = dstpath + '\\' + os.path.basename(srcpath) + 'delcomment'
if not os.path.exists(dstpath):
os.mkdir(dstpath)
print 'dstpath: ' + dstpath

#开始转换
convert(srcpath, dstpath)