码迷,mamicode.com
首页 > 微信 > 详细

代码统计小程序

时间:2019-10-04 23:06:55      阅读:150      评论:0      收藏:0      [点我收藏+]

标签:image   利用   cmd   tin   start   pre   统计   ogr   file   

关于os.work

import os
res = os.walk(r'D:\LearnPython\program2')
#for path, _, file_list in res:
    #print(path)#提取出根目录下所有的文件夹
    #print(file_list)#提取出文件夹里所有的文件
for _,i,_ in res:
    print(i)#一层一层打印出所有文件夹的名字

关于 sys.argv

技术图片

在cmd中运行时如果.py文件中有sys.argv,则可得到一个列表,且列表的第一个元素是执行文件的文件名,后面几个元素是我们在.py旁边输入的字符,以空格分开为每个元素,所以如果编写代码统计的程序,我们可以利用sys.argv的这个特点。Sys.argv[ ]其实就是一个列表,里边的项为用户输入的参数,关键就是要明白这参数是从程序外部输入的。

代码统计程序

此程序只可在cmd下运行

import os
import sys
def count_code_lines(filename):
    res = os.walk(filename)
    #print(res)
    count = 0
    for path, _, file_list in res:#提取出我们输入的文件夹里面的所有文件夹地址和文件名
        print(path)#文件夹地址
        print(file_list)#文件名
        for file in file_list:
            filename = os.path.join(path, file)#拼接出文件的路径
            if filename.endswith('py'):#识别哪些是python文件
                with open(filename, 'r', encoding='utf8') as fr:
                    file_count = 0
                    for i in fr:
                        if i.startswith('#') or i.startswith('\n'):#不计算注释和换行
                            continue
                        count += 1
                        file_count += 1
                    print(f'{filename}有{file_count}行')

    print(f'总共有{count}行')

if __name__ == '__main__':
    filename = sys.argv[1]#当运行cmd的时候直接将要统计代码的地址复制在本程序名旁边即可
    #filename = '目录or文件地址'
    #count_code_lines(r'D:\LearnPython\program2\10.03')
    count_code_lines(filename)

代码统计小程序

标签:image   利用   cmd   tin   start   pre   统计   ogr   file   

原文地址:https://www.cnblogs.com/ghylpb/p/11623292.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!