前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >python实现的代码行数统计代码

python实现的代码行数统计代码

原创
作者头像
大师级码师
修改2021-11-03 12:59:59
1.1K0
修改2021-11-03 12:59:59
举报
文章被收录于专栏:大师级码师大师级码师
代码语言:javascript
复制
'''
      Author: liupengfei
    Function: count lines of code in a folder iteratively
Shell-format: cmd [dir]
   Attention: default file encode is utf8 and default file type is java-source-file. But users can customize this script by just modifing global variables.
'''
import sys
import os
import codecs
from _pyio import open
totalCount = 0;
fileType = '.java'
descLineBegin = '//'
descBlockBegin = r'/**'
descBlockEnd = r'*/'
fileEncode = 'utf-8'
def main():
    DIR = os.getcwd()
    if len(sys.argv) >= 2:
        DIR = sys.argv[1]
    if os.path.exists(DIR) and os.path.isdir(DIR):
        print('target directory is %s' % DIR)
        countDir(DIR)
        print('total code line is %d' % totalCount)
    else:
        print('target should be a directory!')
def isFileType(file):
    return len(fileType) + file.find(fileType) == len(file)
def countDir(DIR):
    for file in os.listdir(DIR):
        absPath = DIR + os.path.sep + file;
        if os.path.exists(absPath):
            if os.path.isdir(absPath):
                countDir(absPath)
            elif isFileType(absPath):
                try:
                    countFile(absPath)
                except UnicodeDecodeError:
                    print(
                        '''encode of %s is different, which
is not supported in this version!'''
                        )
def countFile(file):
    global totalCount
    localCount = 0
    isInBlockNow = False
    f = codecs.open(file, 'r', fileEncode);
    for line in f:
        if (not isInBlockNow) and line.find(descLineBegin) == 0:
            pass;
        elif (not isInBlockNow) and line.find(descBlockBegin) >= 0:
            if line.find(descBlockBegin) > 0:
                localCount += 1
            isInBlockNow = True;
        elif isInBlockNow and line.find(descBlockEnd) >= 0:
            if line.find(descBlockEnd) + len(descBlockEnd) < len(line):
                localCount += 1
            isInBlockNow = False;
        elif (not isInBlockNow) and len(line.replace('\\s+', '')) > 0:
            localCount += 1
    f.close()
    totalCount += localCount
    print('%s : %d' % (file, localCount))
if __name__ == '__main__':
    main();

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档