python实现代码统计程序

Python实现代码统计程序

代码统计程序可以帮助开发人员快速了解自己编写的代码的量、质量等信息，常用于代码管理、项目评估等方面。Python作为一种高级编程语言，拥有丰富的标准库和第三方库，可以轻松实现代码统计程序。

以下是实现代码统计程序的完整攻略：

1.确定需求

首先，需要明确代码统计程序的需求，包括要统计哪些信息、支持哪些类型的文件等。

常见的代码统计信息包括：

• 代码行数
• 空白行数
• 注释行数
• 函数个数
• 类个数

支持的文件类型可能包括已有的代码文件类型，如.py、.java、.c等，也可根据具体需求自定义。

2.搜索文件

根据需求，遍历工程目录下的所有文件，过滤出待统计的代码文件。

参考代码：

def get_files(directory):
    for dirpath, _, filenames in os.walk(directory):
        for filename in filenames:
            if is_code_file(filename):
                yield os.path.join(dirpath, filename)

其中，os.walk方法可对目录进行深度优先遍历，is_code_file方法判断文件是否为待统计的代码文件。

3.统计信息

遍历每个代码文件，统计该文件的相关信息，如代码行数、注释行数、空白行数等，并将这些信息累加到总数中。

参考代码：

def count_file(filename):
    with open(filename, 'r', encoding='utf-8') as f:
        lines = f.readlines()
        code_lines = 0
        blank_lines = 0
        comment_lines = 0
        in_comment = False
        for line in lines:
            line = line.strip()
            if line.startswith('#'):
                comment_lines += 1
            elif line.startswith('"""') or line.startswith("'''"):
                comment_lines += 1
                in_comment = not in_comment
            elif in_comment:
                comment_lines += 1
            elif not line:
                blank_lines += 1
            else:
                code_lines += 1
        return code_lines, blank_lines, comment_lines

4.输出结果

最后，将统计信息输出，可以以表格、柱状图等形式呈现。

参考代码：

def print_table(data):
    headers = ['File', 'Code Lines', 'Blank Lines', 'Comment Lines']
    row_format = "{:<20} {:<12} {:<12} {:<12}"
    print(row_format.format(*headers))
    for filename, code_lines, blank_lines, comment_lines in data:
        print(row_format.format(filename, code_lines, blank_lines, comment_lines))

def main(directory):
    data = []
    for filename in get_files(directory):
        code, blank, comment = count_file(filename)
        data.append((filename, code, blank, comment))
    print_table(data)

示例说明

• 示例一

统计一个Python工程中的代码文件信息。

```python

main('path/to/project')
File Code Lines Blank Lines Comment Lines
path/to/project/foo.py 100 20 50
path/to/project/bar.py 200 30 80
...
```

• 示例二

统计一个C语言工程中的代码文件信息。

```python

main('path/to/c/project')
File Code Lines Blank Lines Comment Lines
path/to/c/project/foo.c 500 50 100
path/to/c/project/bar.c 700 80 150
...
```

通过以上攻略，可以轻松实现一个简单的代码统计程序，用于工程管理、项目评估等方面。