python统计代码函数

import sys
import os

exts = ['.cpp','.h']
def read_line_count(fname):
    count = 0
    for file_line in open(fname).xreadlines():
        count += 1
    return count

if __name__ == '__main__':

    count = 0
    fcount = 0
    for root,dirs,files in os.walk(os.getcwd()):
        for f in files:
            # Check the sub directorys
            fname = (root + '/'+ f).lower()
            ext = f[f.rindex('.'):]
            try:
                if(exts.index(ext) >= 0):
                    fcount += 1
                    c = read_line_count(fname)
                    count += c
            except:
                pass

    print 'file count:%d' % fcount
    print 'count:%d' % count

上面代码是py2的，下面是py3的

import sys
import os
 
exts = ['.cc']
def read_line_count(fname):
    count = 0
    for file_line in open(fname).readlines():
        count += 1
    return count
 
if __name__ == '__main__':
    count = 0
    fcount = 0
    for root,dirs,files in os.walk(os.getcwd()):
        for f in files:
            # Check the sub directorys
            fname = (root + '/'+ f).lower()
            ext = f[f.rfind('.'):]
            try:
                if(exts.index(ext) >= 0):
                    fcount += 1
                    c = read_line_count(fname)
                    count += c
            except:
                pass
 
    print('file count:%d' % fcount)
    print('count:%d' % count)