用Python来实现一个代码行数统计工具

最新推荐文章于 2022-09-06 10:31:56 发布

原创

最新推荐文章于 2022-09-06 10:31:56 发布 · 695 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Python #后端 #编程语言

本文介绍如何使用Python创建一个代码行数统计工具，该工具能统计每个文件的行数，总行数，运行时间，并支持指定文件类型，递归统计文件夹及其子文件夹，还能排除空行。此外，文章还鼓励读者在此基础上扩展功能，如排除注释行。

我们经常想要统计项目的代码行数，但是如果想统计功能比较完善可能就不是那么简单了，今天我们来看一下如何用python来实现一个代码行统计工具。

思路：首先获取所有文件，然后统计每个文件中代码的行数，最后将行数相加.

实现的功能：

统计每个文件的行数；

统计总行数；

统计运行时间；

支持指定统计文件类型，排除不想统计的文件类型；

递归统计文件夹下包括子文件件下的文件的行数；

排除空行；

# coding=utf-8
import os
import time
basedir = '/root/script'
filelists = []
# 指定想要统计的文件类型
whitelist = ['php', 'py']
#遍历文件, 递归遍历文件夹中的所有
def getFile(basedir):
    global filelists
    for parent,dirnames,filenames in os.walk(basedir):
        #for dirname in dirnames:
        #    getFile(os.path.join(parent,dirname)) #递归
        for filename in filenames:
            ext = filename.split('.')[-1]
            #只统计指定的文件类型，略过一些log和cache文件
            if ext in whitelist:
                filelists.append(os.path.join(parent,filename))
#统计一个文件的行数
def countLine(fname):
    count = 0
    for file_line in open(fname).xreadlines():
        if file_line != '' and file_line != '\n': #过滤掉空行
            count += 1
    print fn