记录开发代码量的小程序

　　这段时间在开发网站,想要去记录自己的开发代码量,所以写了一个小脚本,用来记录,代码如下:

#-*- coding: UTF-8 -*-
#用来检测当前的代码量
import sys,os
#全局变量,记录当前的代码行数
count = 0
#根据文件名称指定文件读取策略,可以选择读取哪些文件或者跳过某些文件,默认跳过所有文件,只读取指定文件.
def fileCheckStrategy(file):
    if ".pyc" in file:
        return False
    if ".py" in file:
        return True
    return False
#目录读取策略,同文件读取策略,用于跳过某些目录,默认目录都遍历
def dirCheckStrategy(dir):
    return True

#读取指定文件的代码行数
def readfile(file):
    global count
    f = open(file,'r')
    read = f.readlines()
    for i in read:
        count+=1
    f.close()

#遍历目录算法
def check(currentpath):
    s = os.listdir(currentpath)
    os.chdir(currentpath)
    for x in s:
        os.chdir(currentpath)
        if os.path.isdir(x):
            if dirCheckStrategy(x):
                #递归遍历指定目录
                print "DIR   "+x
                check(currentpath+"\"+x)
        elif os.path.isfile(x):
            if fileCheckStrategy(x):
                #读取指定文件的代码行数
                print "FILE  "+x
                readfile(currentpath+"\"+x)

    pass

if __name__ == '__main__':
    #a = sys.argv[0]
    #从当前工作目录开始,按照策略读取代码量
    path = os.getcwd()
    print path
    check(path)
    print count

主要是用到一个递归来设计,算法较为简单  就不多写了