1import os
 2import time
 3basedir = os.path.dirname(__file__)
 4filelists = []
 5# 指定想要统计的文件类型
 6whitelist = ['php']
 7#遍历文件, 递归遍历文件夹中的所有
 8def getFile(basedir):
 9    global filelists
10    for parent,dirnames,filenames in os.walk(basedir):
11        #for dirname in dirnames:
12        #    getFile(os.path.join(parent,dirname)) #递归
13        for filename in filenames:
14            ext = filename.split('.')[-1]
15            #只统计指定的文件类型，略过一些log和cache文件
16            if ext in whitelist:
17                filelists.append(os.path.join(parent,filename))
18#统计一个文件的行数
19def countLine(fname):
20    count = 0
21    single_quotes_flag = False
22    double_quotes_flag = False
23    fname_new = fname+'.new.txt'
24    outfile = open(fname_new,'wb')
25 
26    with open(fname, 'rb') as f:
27        for file_line in f:
28            file_line_old=file_line
29            file_line = file_line.strip()
30            # print(file_line)
31            # 空行
32            if file_line == b'':
33                pass
34 
35            # 注释 # 开头
36            elif file_line.startswith(b'#'):
37                pass
38 
39            # 注释 // 开头
40            elif file_line.startswith(b'//'):
41                pass
42 
43            # 注释/*  */在同一行开头结尾的情况
44            elif file_line.startswith(b"/*") and file_line.endswith(b"*/"):
45                single_quotes_flag = False
46 
47 
48            # 注释 单引号 ''' 开头---/*
49            elif file_line.startswith(b"/*") and not single_quotes_flag:
50                single_quotes_flag = True
51            # 注释 中间 和 ''' 结尾---- */
52            elif single_quotes_flag == True:
53                if file_line.endswith(b"*/"):
54                    single_quotes_flag = False
55 
56            # 注释 双引号 """ 开头
57            #elif file_line.startswith(b'"""') and not double_quotes_flag:
58            #    double_quotes_flag = True
59            # 注释 中间 和 """  结尾
60            #elif double_quotes_flag == True:
61            #    if (file_line.endswith(b'"""')):
62            #        double_quotes_flag = False
63 
64            # 代码
65            else:
66                #outfile.write(file_line_old+b'\n')
67                #str=file_line.strip().replace(b' ', b'').replace(b'\n', b'').replace(b'\t', b'').replace(b'\r', b'').strip()
68                #if (not str2.startswith(b'//') and str2!="\n"):
69                outfile.write(file_line_old)
70                count += 1
71 
72        print(fname + '----', count)
73        outfile.close()
74        # 单个文件行数
75        # print(fname,'----count:',count)
76        return count
77 
78if __name__ == '__main__' :
79    startTime = time.clock()
80    getFile(basedir)
81    totalline = 0
82    for filelist in filelists:
83        totalline = totalline + countLine(filelist)
84    print('\033[43m total lines: \033[0m'.center(20,'-'),totalline)
85    print('Done! Cost Time: %0.5f second' % (time.clock() - startTime))