採集容器記憶體並寫到excel

標籤：stream   %s   exec   ret   $1   node   os.path   get   bsp   

 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 output13 14 15 def get_docker_name():16     infos = execute("docker ps |awk ‘{print $1, $NF}‘").split(‘\n‘)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_name22     return id_name23 24 25 def get_docker_mem():26     regex = re.compile(‘\s+‘)27     ret = execute(‘docker stats --no-stream‘).split(‘\n‘)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 = 136     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) * 102443         if unit.startswith(‘K‘):44             mem = float(mem) / 102445         try:46             mem = float(mem)47         except:48             pass49         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 += 154     w.save(result_name)55 56 57 if __name__ == ‘__main__‘:58     get_docker_mem()

 

採集容器記憶體並寫到excel