1 # coding=utf-8 2 import os 3 import commands 4 import re 5 from pyExcelerator import * 6 7 8 def execute(cmd): 9 status, output = commands.getstatusoutput(cmd) 10 if status != 0: 11 raise Exception('status is %s, output is %s' % (status, output)) 12 return output 13 14 15 def get_docker_name(): 16 infos = execute("docker ps |awk '{print $1, $NF}'").split(' ') 17 regex = re.compile('s+') 18 id_name = {} 19 for info in infos: 20 docker_id, docker_name = regex.split(info) 21 id_name[docker_id] = docker_name 22 return id_name 23 24 25 def get_docker_mem(): 26 regex = re.compile('s+') 27 ret = execute('docker stats --no-stream').split(' ') 28 result_name = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'docker_res.xlsx') 29 id_name = get_docker_name() 30 w = Workbook() 31 ws = w.add_sheet('node_1_data') 32 ws.write(0, 0, 'docker_id') 33 ws.write(0, 1, 'docker_name') 34 ws.write(0, 2, 'mem(MB)') 35 index = 1 36 for docker in ret: 37 info = regex.split(docker) 38 docker_id = info[0] 39 mem = info[2] 40 unit = info[3] 41 if unit.startswith('G'): 42 mem = float(mem) * 1024 43 if unit.startswith('K'): 44 mem = float(mem) / 1024 45 try: 46 mem = float(mem) 47 except: 48 pass 49 name = id_name[docker_id] 50 ws.write(index, 0, docker_id) 51 ws.write(index, 1, name) 52 ws.write(index, 2, mem) 53 index += 1 54 w.save(result_name) 55 56 57 if __name__ == '__main__': 58 get_docker_mem()