您的位置:首页 > 编程语言 > Python开发

Python:递归遍历目录,统计目录大小,并以列表的形式返回所有py文件

2017-10-13 14:32 681 查看
# 递归遍历目录,统计目录大小,并以列表的形式返回所有py文件

import os
path = 'F:/qianfeng/python36/day08/nian'
dicList = []

def myTotalSize(path):
# 全局变量dicList,存放py文件名
global dicList

# 存放目录字节数
memory = 0

# 当前目录下的文件列表
fileList = os.listdir(path)

for file in fileList:  # 以列表的形式返回该目录下的所有文件
# print(file)
newPath = os.path.join(path, file)  # 将每一个文件拼接成绝对路径
# print(newPath)

# 判断是否为目录
if os.path.isdir(newPath):
memory += myTotalSize(newPath)  # 累加每个目录的大小

# 判断是否为文件
elif os.path.isfile(newPath):
# 判断是否为py文件
if os.path.splitext(newPath)[1].upper() == '.PY':
dicList.append(os.path.basename(newPath))
memory += os.path.getsize(newPath)  # 累加每个文件的大小

# 返回文件大小
return memory

# 打印目录大小(总字节数)
print(myTotalSize(path))

# 打印py文件列表
print(dicList)
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  python 函数 递归 path