Java代码统计工具-Python版

た入场券 2022-11-15 05:47 178阅读 0赞

### 前序 ###

前两天工作需要用java写了一个Java工程代码量统计工具，最近在学习python,想着锻炼一下，把java代码翻译成python。

### 使用方法 ###

# -e 文件扩展名
    # -f 要统计代码量的文件夹
    py StatisticCodeLines.py -e java -f d:\test

#!/usr/bin/env python
    # -*- encoding: utf-8 -*-
    ''' @File : StatisticCodeLines.py @Time : 2021/03/30 17:05:25 @Author : Liang Haijiang @Version : 1.0 @Contact : lianghj@geovis.com.cn @Desc : None '''
    
    # here put the import lib
    import os
    import argparse
    import re
    
    #正常代码
    global NORMAL_LINES
    NORMAL_LINES = 0
    #空白行代码
    global WHILE_LINES
    WHILE_LINES = 0
    #注释代码
    global COMMENT_LINES
    COMMENT_LINES = 0
    
    def statisticCode(args):
        extName = args.ext
        path = args.folder
        if os.path.exists(path):
            traverseFolder(path,extName)
        total = NORMAL_LINES + WHILE_LINES + COMMENT_LINES
        print('有效行数:' + str(NORMAL_LINES))
        print('空白行数:' + str(WHILE_LINES))
        print('注释行数:' + str(COMMENT_LINES))
        print('总计行数:' + str(total))
    
    def traverseFolder(path,extName):
        if os.path.isdir(path):
            files = os.listdir(path)
            for file in files:
                childPath = os.path.join(path,file)
                traverseFolder(childPath,extName)
        if os.path.isfile(path) and path.endswith(extName):
            parseFile(path)
    
    def parseFile(path):
        global COMMENT_LINES
        global WHILE_LINES
        global NORMAL_LINES
    
        comment = False
        tempWhiteLines = 0
        tempCommentLines = 0
        tempNormalLines = 0
    
        fo = open(path,"r",encoding='UTF-8')
        lines = fo.readlines()
        for line in lines:
            line = line.strip()
            if len(line) == 0:
                tempWhiteLines += 1
            elif line.startswith('/*'):
                tempCommentLines += 1
                comment = not line.endswith('*/')
            elif comment and not line.endswith('*/'):
                 tempCommentLines += 1
            elif comment and line.endswith('*/'):
                tempCommentLines += 1
                comment = False
            elif line.startswith('//'):
                tempCommentLines += 1
            else:
                tempNormalLines += 1
        fo.close()
    
        print(path + '::' + str(tempCommentLines) + ':' + str(tempNormalLines) + ':' + str(tempWhiteLines))
        WHILE_LINES += tempWhiteLines
        COMMENT_LINES += tempCommentLines
        NORMAL_LINES += tempNormalLines
    if __name__ == "__main__":
        argpaarser = argparse.ArgumentParser()
        argpaarser.add_argument('-e','--ext',help=u'文件后缀名')
        argpaarser.add_argument('-f','--folder',help=u'统计文件夹')
        argpaarser.set_defaults(func = statisticCode)
        args = argpaarser.parse_args()
        args.func(args)