CMDB 资产采集——插件可插拔式、可拓展思想

PLUGIN_DICT = {
    ‘disk‘:‘src.plugins.disk.Disk‘,                 #对应value 为采集disk 的类的路径
    ‘memory‘:‘src.plugins.memory.Memory‘,
    ‘network‘:‘src.plugins.network.Network‘,
}

from .base import BaseHandler
from ..plugins import get_server_info
class AgentHandler(BaseHandler):

    def cmd(self,command,hostname=None):
        import subprocess
        return subprocess.getoutput(command)        #command 为资产采集的命令例如内存采集“free -m”,采集后的数据通过getouput 收集

    def handler(self):
        """
        处理Agent模式下的资产采集：网卡、内存、硬盘
        :return:
        """
        # 通过调用get_server_info获取所有的资产信息：网卡、内存、硬盘,把AgentHandler 类通过self 作为参数传递给get_server_info最终传递给资产采集插件
        info = get_server_info(self)
        print(‘agent‘,info)

from config import settings
from .base import BaseHandler
from ..plugins import get_server_info

class SSHHandler(BaseHandler):
    #cmd 函数为ssh 模式的远程登录命令，需要远程登录的主机名与登陆后执行的命令作为参数
    def cmd(self, command, hostname=None):
        """
        调用paramiko远程连接主机并执行命令，依赖rsa
        :param hostname:主机名
        :param command: 要执行的命令
        :return:
        """
        import paramiko

        private_key = paramiko.RSAKey.from_private_key_file(settings.SSH_PRIVATE_KEY)
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(hostname=hostname, port=settings.SSH_PORT, username=settings.SSH_USER, pkey=private_key)
        stdin, stdout, stderr = ssh.exec_command(command)
        result = stdout.read()
        ssh.close()
        return result
    #
    def handler(self):
        """
        处理SSH模式下的资产采集
        :return:
        """
        print(‘ssh模式‘,get_server_info(self))

from config import settings
from lib.module_string import import_string

def get_server_info(handler,hostname=None):                 #hostname=None 默认为空，传递参数时可以不传递，也不会报错
    """
    循环所有的插件，获取所有的资产信息，然后返回
    :param handler:
    :return:
    """
    info = {}
    for name,path in settings.PLUGIN_DICT.items():
        cls = import_string(path)                           #import_string 为通过importlib getattr 导入类的函数
        obj = cls()
        # obj 为资产采集插件，每个插件都有一个process 方法进行资产采集；handler 为AgentHandler/SSHHandler类
        result = obj.process(handler,hostname)
        info[name] = result                                 #{‘disk‘:100G,‘memory‘:16G,‘network‘:xxxxxx}

    return info

class Disk(object):
    def process(self,handler,hostname):
        """
        获取硬盘信息
        :return:
        """
        result = handler.cmd(‘dir‘,hostname)    #handler 为模式引擎的类AgentHandler/SaltHandler
        return result

class Memory(object):
    def process(self, handler, hostname):
        """
        获取内存信息
        :return:
        """
        result = handler.cmd(‘wmic memorychip‘, hostname)   #handler 为模式引擎的类AgentHandler/SaltHandler
        return result