import os

import time

from pathlib import Path

def traverse_directory(path, indent=0, file_stats=None):

“””

递归遍历目录并打印目录树结构

参数:

path: 要遍历的目录路径

indent: 缩进级别（用于打印树状结构）

file_stats: 文件类型统计字典

“””

if file_stats is None:

file_stats = {}

# 获取目录名

dir_name = os.path.basename(path) or path

# 打印当前目录（带缩进）

print(” ” * indent + f” {dir_name}/”)

try:

# 列出目录内容

with os.scandir(path) as entries:

for entry in entries:

# 计算下一级缩进

next_indent = indent + 4

if entry.is_dir():

# 递归遍历子目录

traverse_directory(entry.path, next_indent, file_stats)

elif entry.is_file():

# 处理文件

file_ext = os.path.splitext(entry.name)[1].lower() or “无扩展名”

# 更新文件统计

if file_ext not in file_stats:

file_stats[file_ext] = {“count”: 0, “size”: 0}

file_stats[file_ext][“count”] += 1

file_stats[file_ext][“size”] += entry.stat().st_size

# 打印文件信息

size_kb = entry.stat().st_size / 1024

mod_time = time.strftime(“%Y-%m-%d %H:%M”, time.localtime(entry.stat().st_mtime))

print(” ” * next_indent + f” {entry.name} ({size_kb:.2f} KB, 修改于 {mod_time})”)

except PermissionError:

print(” ” * next_indent + “⚠️ 权限不足，无法访问”)

except Exception as e:

print(” ” * next_indent + f”❌ 错误: {str(e)}”)

return file_stats

def print_summary(file_stats):

“””打印文件类型统计摘要”””

print(”
文件类型统计:”)

print(“-” * 50)

print(f”{'扩展名':<15} {'数量':<10} {'总大小(KB)':<15} {'平均大小(KB)':<15}”)

print(“-” * 50)

total_files = 0

total_size = 0

for ext, stats in sorted(file_stats.items(), key=lambda x: x[1]['size'], reverse=True):

count = stats[“count”]

size_bytes = stats[“size”]

size_kb = size_bytes / 1024

avg_size = size_kb / count if count > 0 else 0

print(f”{ext:<15} {count:<10} {size_kb:<15.2f} {avg_size:<15.2f}”)

total_files += count

total_size += size_kb

print(“-” * 50)

print(

f”{'总计':<15} {total_files:<10} {total_size:<15.2f} {total_size / total_files if total_files > 0 else 0:<15.2f}”)

print(“-” * 50)

def main():

# 获取用户输入的目录路径

default_path = os.getcwd()

user_input = input(f”请输入要遍历的目录路径 (默认为当前目录 '{default_path}'): “).strip()

target_dir = user_input if user_input else default_path

# 检查目录是否存在

if not os.path.isdir(target_dir):

print(f”错误: 目录 '{target_dir}' 不存在!”)

return

print(f”
开始遍历目录: {os.path.abspath(target_dir)}
“)

# 遍历目录并收集统计信息

file_stats = traverse_directory(target_dir)

# 打印统计摘要

print_summary(file_stats)

print(”
✅ 遍历完成!”)

if __name__ == “__main__”:

main()