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' % memory
    else:
        memory = i / 1024. / 1024./ 1024.
        print '%.4f GB' % memory

2、检测指定路径剩余储存空间大小

import ctypes
import os
import platform
import sys


def get_free_space_mb(folder):
    """ Return folder/drive free space (in bytes)
    """
    if platform.system() == 'Windows':
        free_bytes = ctypes.c_ulonglong(0)
        ctypes.windll.kernel32.GetDiskFreeSpaceExW(ctypes.c_wchar_p(folder), None, None, ctypes.pointer(free_bytes))
        return free_bytes.value/1024/1024/1024 
    else:
        st = os.statvfs(folder)
        return st.f_bavail * st.f_frsize/1024/1024/1024.

这个适用于unix系统下，windows系统下 os 无 statvfs 属性。

def disk_stat(path):
    import os
    hd={}
    disk = os.statvfs(path)
    percent = (disk.f_blocks - disk.f_bfree) * 100 / (disk.f_blocks -disk.f_bfree + disk.f_bavail) + 1
    return percent
print disk_stat('.')