统计python代码行数

Posted 2020-10-14

    addhosts项目已接近尾声，我很想知道我们写了多少行代码。

一、需求

    统计源码目录下py文件的代码行数。

    

二、脚本分析

    获取指定目录下所有的.py文件，对文件进行遍历；

    读取每个文件，对文件内容进行遍历，过滤掉空行和注释；

三、实现及结果

#coding:utf-8
import os

class StatLines(object):

    def __init__(self,path):
        self.path = path

    def stat_lines(self):
        file_list = os.listdir(self.path)
        os.chdir(self.path)
        total = 0
        for file in file_list:
            if file.endswith(‘.py‘):
                lines = open(file, encoding=‘utf-8‘).readlines()
                count = 0
                for line in lines:
                    if line == ‘\n‘:
                        continue
                    elif line.startswith(‘#‘):
                        continue
                    else:
                        count += 1
                total += count
                print(‘%s has %d lines‘ %(file,count))
        print(‘total lines is: %d‘ %total)

if __name__ == ‘__main__‘:
    sl = StatLines(‘E:\\Python_Project\\addhost_v2\\addhosts‘)
    sl.stat_lines()

运行结果如下：

四、总结

    问题：

    在执行open(file).readlines()时，遇到了如下错误

“UnicodeDecodeError: ‘gbk‘ codec can‘t decode byte 0xaf in position 548: illegal multibyte sequence”

    解决方法：

    在open时，设置encoding=‘utf-8‘后，问题得到解决。

本文出自 “乐学园” 博客，请务必保留此出处http://2338053.blog.51cto.com/2328053/1982133