python--代码统计（进阶版）

　　在上一篇的随笔中发表了代码统计小程序，但是发表后，我发现，以前写的代码怎么办

　　写了那么多，怎么就从0开始了呢，，，，我还是个孩子啊，不能这么残忍

　　于是，代码统计进阶版：统计当前目录下所有指定文件类型的行数

　　

#coding:gbk
import os
import time
from CountItem.FindCode import *

n = 0
'''查找历史记录的行数'''
try:
    with open('TotalLines','r') as p:
        lastline = ''
        for lastline in p.readlines():
            pass
        index = lastline.find('>>')
        n = int(lastline[index+2:])
except ValueError as e:
    n = 0
except FileNotFoundError:
    n = 0
'''文件列表'''
fileList = FindCode().getLocalfile(os.getcwd())
'''计算行数'''
for filename in fileList:
    try:
        with open(filename,'r') as f:
            try:
                lines = f.readlines()
            except UnicodeDecodeError:
                '''编码错误，不用管，我们要的是行数'''
                '''嗯，，掩耳盗铃'''
                pass
            for s in lines:
                '''不计入空行'''
                if s == '
':
                    continue
                n += 1
    except FileNotFoundError as e:
        print('文件名或文件路径错误，没有该文件！')
        os.system('pause')
        os._exit(1)
    except OSError as e:
        print('文件名不合法')
        os.system('pause')
        os._exit(1)

print(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())) + '代码行数>>' + str(n) + '
')

'''写入文件'''
with open('TotalLines','a') as f:
    f.write(time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())) + '代码行数>>' + str(n) + '
')
os.system('pause')

#coding:gbk
import os
class FindCode():
    '''初始化查找的文件类型列表'''
    def __init__(self):
        self.targetFile = []
        self.targetFile.append('c')
        self.targetFile.append('cpp')
        self.targetFile.append('py')
        self.targetFile.append('java')
    '''递归查找文件,用绝对路径'''
    def getLocalfile(self,findpath):
        fileList = os.listdir(findpath) 
        aimsList = []
        for filepath in fileList:
            filepath = findpath + '\' + filepath
            if os.path.isdir(filepath):
                aimsList.extend(self.getLocalfile(filepath))
            else:
                if self.selectFile(filepath):
                    aimsList.append(filepath)
        return aimsList
    
    def selectFile(self,filepath):
        index = filepath.find('.')
        lastname = filepath[index+1:]
        return True if lastname in self.targetFile else False
    
if __name__ == '__main__':
    Demo = FindCode()
    print('
'.join(x for x in Demo.getLocalfile(os.getcwd())))
    

但是，不能统计一次挪一次窝吧，再改！

'''文件列表'''
filepath = input('请输入指定文件路径（绝对路径）：')
if os.path.isdir(filepath):
     fileList = FindCode().getLocalfile(filepath)

这个就对了，很爽，，，