正文

如何用Python脚本轻松统计所有Python文件的大小，快速找出“身材庞大”的程序代码

/0 浏览量
1020

在软件开发和维护过程中，了解各个Python文件的大小可以帮助我们更好地管理代码库，尤其是在处理大型项目时。以下是一个简单的Python脚本，它可以帮助你统计指定目录下所有Python文件的大小，并快速找出“身材庞大”的程序代码。

准备工作

在开始之前，请确保你的环境中已安装了Python。以下脚本适用于Python 3.x版本。

脚本说明

这个脚本会递归地遍历指定目录，统计所有Python文件（以.py结尾的文件）的大小，并打印出每个文件的大小以及总大小。

代码实现

import os

def get_size(start_path = '.'):
    total_size = 0
    for dirpath, dirnames, filenames in os.walk(start_path):
        for f in filenames:
            fp = os.path.join(dirpath, f)
            if os.path.exists(fp):
                total_size += os.path.getsize(fp)
    return total_size

def find_large_py_files(directory, size_threshold):
    large_files = []
    total_size = get_size(directory)
    print(f"Total size of Python files in '{directory}': {total_size} bytes")
    
    for dirpath, dirnames, filenames in os.walk(directory):
        for filename in filenames:
            if filename.endswith('.py'):
                filepath = os.path.join(dirpath, filename)
                if os.path.getsize(filepath) > size_threshold:
                    large_files.append((filepath, os.path.getsize(filepath)))
    return large_files

# 使用示例
directory_to_search = '/path/to/your/directory'  # 修改为你要搜索的目录
size_threshold = 1024 * 1024  # 1MB
large_py_files = find_large_py_files(directory_to_search, size_threshold)

print("\nLarge Python files:")
for filepath, size in large_py_files:
    print(f"{filepath}: {size} bytes")

使用说明

  1. 将directory_to_search变量修改为你想要搜索的目录路径。
  2. 将size_threshold变量修改为你认为文件大小超过这个值就属于“身材庞大”的值。
  3. 运行脚本，它会输出所有Python文件的大小，并突出显示超过大小阈值的文件。

注意事项

  • 脚本中使用的os.walk()函数会递归遍历所有子目录。
  • 脚本中的os.path.getsize()函数用于获取单个文件的大小。
  • 脚本假设所有以.py结尾的文件都是Python文件，如果你有其他类型的文件也以.py结尾，你可能需要调整脚本来排除它们。

通过这个脚本，你可以轻松地找到你的代码库中“身材庞大”的Python文件，从而进行优化或重构。

-- 展开阅读全文 --