import os

import re

from .base import BasePlugin

from lib.response import BaseRespone

class Disk(BasePlugin):

def win(self, handler, hostname):

ret = handler.cmd("hostname", hostname)

response = BaseResponse()

return response.dict

def linux(self, handler, hostname):

response = BaseRespone()

try:

if self.debug:

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

print(output)

else:

shell_command = "sudo MegaCli -PDList -aALL"

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: 解析后的结果

"""

response = {}

result = []

for row_line in content.split("\n\n\n\n"):

result.append(row_line)

for item in result:

temp_dict = {}

for row in item.split("\n"):

if not row.strip():

continue

if len(row.split(":")) != 2:

continue

key, value = row.split(":")

name = self.mega_patter_match(key)

if name:

if key == 'Raw Size':

raw_size = re.search('(\d+\.\d+)', value.strip())

if raw_size:

temp_dict[name] = raw_size.group()

else:

raw_size = '0'

else:

temp_dict[name] = value.strip()

if temp_dict:

response[temp_dict['slot']] = temp_dict

return response

@staticmethod

def mega_patter_match(needle):

grep_pattern = {'Slot': 'slot', 'Raw Size': 'capacity', 'Inquiry': 'model', 'PD Type': 'pd_type'}

for key, value in grep_pattern.items():

if needle.startswith(key):

return value

return False