通过给定目录，统计所有的不同子文件类型及占用内存

#!/usr/bin/env python

# encoding: utf-8

"""

@author: wayne

@file: file_type.py

@time: 2018/11/13 19:46

"""

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= "E:\\联营项目资料库"

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)))