from .base import BasePlugin

import os

from lib import convert

from lib.response import BaseResponse

class Memory(BasePlugin):

def win(self, handler, hostname):

"""

执行命令拿到结果

:return:

"""

ret = handler.cmd('getmac', hostname)

response = BaseResponse()

return response.dict

def linux(self, handler, hostname):

response = BaseResponse()

try:

if self.debug:

output = open(os.path.join(self.base_dir, 'files', 'memory.out'), 'r').read()

else:

shell_command = "sudo dmidecode -q -t 17 2>/dev/null"

output = handler.cmd(shell_command, hostname)

response.data = self.parse(output)

except Exception as e:

msg = traceback.format_exc()

response.status = False

response.error = msg

logger.error(msg)

return response.dict

def parse(self, content):

"""

解析shell命令返回结果

:param content: shell 命令结果

:return:解析后的结果

"""

ram_dict = {}

key_map = {

'Size': 'capacity',

'Locator': 'slot',

'Type': 'model',

'Speed': 'speed',

'Manufacturer': 'manufacturer',

'Serial Number': 'sn',

}

devices = content.split('Memory Device')

for item in devices:

item = item.strip()

if not item:

continue

if item.startswith('#'):

continue

segment = {}

lines = item.split('\n\t')

for line in lines:

if len(line.split(':')) > 1:

key, value = line.split(':')

else:

key = line.split(':')[0]

value = ""

if key in key_map:

if key == 'Size':

segment[key_map['Size']] = convert.convert_mb_to_gb(value, 0)

else:

segment[key_map[key.strip()]] = value.strip()

ram_dict[segment['slot']] = segment

return ram_dict