python检测空间储存剩余大小和指定文件夹内存占用

1、检测指定路径下所有文件所占用内存

import os

def check_memory(path, style='M'):
    i = 0
    for dirpath, dirname, filename in os.walk(path):
        for ii in filename:
            i += os.path.getsize(os.path.join(dirpath,ii))
    if style == 'M':
        memory = i / 1024. / 1024.
        print '%.2f MB'