from subprocess import Popen, PIPE

# 执行操作系统命令并返回执行结果

def exec_shell(shell_cmd):

process = Popen(shell_cmd, shell=True, stdout=PIPE, stderr=PIPE)

stdout, stderr = process.communicate()

return stdout, stderr

# 获取磁盘使用率

# 参数 target_path: 待检测挂载点，如果指定了该参数则只检测该挂载点，否则检测所有本地文件系统

def get_disk_space_usage(target_path=''):

data = []

cmd = 'df -lm' # Local FileSystem only

if target_path.strip() != '':

if os.path.exists(target_path):

cmd = 'df -m '+target_path.strip()

res = exec_shell(cmd)[0]

res_lines = res.split(os.linesep)

del res_lines[0] # delete line header

for line in res_lines:

if '%' in line:

line_tmp = re.split( r'\s+|\t', line.strip() )

if '%' in line_tmp[-2] and '/' in line_tmp[-1]:

line_data = {}

line_data['total_mb'] = int( line_tmp[-5].strip() )

line_data['used_mb'] = int( line_tmp[-4].strip() )

line_data['free_mb'] = int( line_tmp[-3].strip() )

line_data['used_pct'] = int( line_tmp[-2].strip().strip('%') )

line_data['free_pct'] = 100 - line_data['used_pct']

line_data['mount_point'] = line_tmp[-1].strip()

data.append(line_data)

return data

# 调用函数

res = get_disk_space_usage()