个人代码量统计工具

代码量能体现一个程序员的能力（起码是正相关的），恰逢最近看到了os模块一些用法，手头无事，随手写了一个小小的工具用来统计个人代码量。

# Author: BaoJunxian
# Date: 3/6/2018

#path format is for windows temporarily

import os

def cal(filePath=None):
    count = 0
    annoFlag = 0
    with open(filePath) as file:
        f = iter(file)
        while True:
            context = next(f,'EOF').strip('\n')
            if context.strip() == 'EOF':
                break
            elif context.strip() == '': #ignore blank line
                continue
            elif context.strip().startswith('#'):
                continue
            elif context.strip().startswith("'''"):
                annoFlag += 1
            elif annoFlag % 2 != 0: #couple annotation
                continue
            else:
                count += 1
        return count

def ergodic(dirPath=None):
    sum = 0
    for eachFile in os.listdir(dirPath):
        if os.path.isdir(dirPath+'\\'+eachFile):
            sum += ergodic(dirPath+'\\'+eachFile)
        elif os.path.isfile(dirPath+'\\'+eachFile):
            if eachFile.endswith('.py') or\
                    eachFile.endswith('.c') or\
                    eachFile.endswith('.cpp'):
                sum += cal(dirPath+'\\'+eachFile)
    return sum

def main():
    DIR = 'C:\\Users\\Stan Pao\\OneDrive\\CodeRepo'
    # manually designate the dir to calculate
    print('All Repo Lines are:',ergodic(DIR),'lines')

if __name__ == '__main__':
    main()
>>>>仍已指定一个源码文件夹
All Repo Lines are: 548 lines

由于本人写python用windows多一点，路径处理方式适配的windows格式，Linux更方便，可以快速改成shell交互指定dir路径。计数规则只简单定义了几条：忽略python两种注释，忽略空行。