#!/usr/bin/env python
# encoding: utf-8
"""
@author: eguotangseng
@file: file_type.py
@time: 2020/02/11
"""
import os
"""
通過給定目錄，統計所有的不同子文件類型及佔用內存
"""
size_dict = {}
type_dict = {}
def get_size_type(path):
files = os.listdir(path)
for filename in files:
temp_path = os.path.join(path, filename)
if os.path.isdir(temp_path):
get_size_type(temp_path) #遞歸
elif os.path.isfile(temp_path):
type_name=os.path.splitext(temp_path)[1]
#無後綴名的文件
if not type_name:
type_dict.setdefault("None", 0)
type_dict["None"] += 1
size_dict.setdefault("None", 0)
size_dict["None"] += os.path.getsize(temp_path)
else:
type_dict.setdefault(type_name, 0)
type_dict[type_name] += 1
size_dict.setdefault(type_name, 0)
size_dict[type_name] += os.path.getsize(temp_path) #獲取文件大小
path= "路徑"
get_size_type(path)
for each_type in type_dict.keys():
print ("該文件夾下共有【 %s 】的文件【 %d 】個 ,佔用內存【 %.2f 】MB" %
(each_type,type_dict[each_type],size_dict[each_type]/(1024*1024)))
print("總文件數：【 %d 】"%(sum(type_dict.values())))
print("總內存大小：【 %.2f 】GB"%(sum(size_dict.values())/(1024**3)))