使用python写的一个代码统计程序

# encoding="utf-8"

"""

统计代码行数

"""

import sys

import os

def count_file_line(path):

    """统计文件的有效行数"""

    countLine = 0

    # 设置一个标志位，当遇到以"""或者'''开头或者结尾的时候，置为False

    flag = True

    # 使用utf-8格式的编码方式读取文件，如果读取失败，将使用gbk编码方式读取文件

    try:

        fp = open(path, "r", encoding="utf-8")

        encoding_type = "utf-8"

        fp.close()

    except:

        encoding_type = "gbk"

    with open(path, "r", encoding=encoding_type) as fp:

        for line in fp:

            # 空行不统计

            if line.strip():

                line = line.strip()

                # 注意下面的这两个elif必须要前面，这样子当('"""')结束之后及时将flag置为True

                if line.endswith('"""') and flag == False:

                    flag = True

                    continue

                if line.endswith("'''") and flag == False:

                    flag = True

                    continue

                if flag == False:

                    continue

                if line.startswith("#!") or line.startswith("#-*-") or line.startswith("# encoding"):

                    countLine += 1

                # 如果以“#”号开头的，不统计

                elif line.startswith("#"):

                    continue

                # 如果同时以("'''")或者('"""')开头或者结尾（比如："""aaa"""），那么不统计

                elif line.startswith('"""') and line.endswith('"""') and line != '"""':

                    continue

                elif line.startswith("'''") and line.endswith("'''") and line != "'''":

                    continue

                # 如果以("'''")或者('"""')开头或者结尾（比如：aaa"""或者"""bbb），那么不统计

                # 注意下面的这两个elif必须要放后面

                elif line.startswith('"""') and flag == True:

                    flag = False

                    continue

                elif line.startswith("'''") and flag == True:

                    flag = False

                    continue

                else:

                    countLine += 1

    return countLine

def count_codes(path,file_types=[]):

    """统计所有文件代码行"""

    # 判断path是目录还是文件，如果是目录的话，遍历目录下所有的文件

    if not os.path.exists(path):

        print("您输入的路径不存在！")

        return 0

    countTotalLine = 0

    file_paths = {}

    if os.path.isdir(path):

        for root,dirs,files in os.walk(path):

            for name in files:

                if not file_types:

                    file_types = ["txt","py"]

                    # print(file_types)

                if os.path.splitext(name)[1][1:] in file_types:

                    file_path = os.path.normpath(os.path.join(root,name))

                    # print(file_path)

                    file_lines = count_file_line(file_path)

                    countTotalLine += file_lines

                    file_paths[file_path] = file_lines

    else:

        if not file_types:

            file_types = ["txt","py"]

        if os.path.splitext(path)[1][1:] in file_types:

            countTotalLine = count_file_line(path)

            file_paths[path] = count_file_line(path)

    return countTotalLine,file_paths

if __name__ == "__main__":

    # 打印出命令行输入的参数

    # print(sys.argv)

    if len(sys.argv) < 2:

        print("请输入路径！")

        sys.exit()

    path = sys.argv[1]

    # print(path)

    file_types = sys.argv[2:]

    # print(file_types)

    print(count_codes(path,file_types))
巴特西

使用python写的一个代码统计程序

最新文章

热门文章
