CMDB学习之三数据采集

判断系统因为是公用的方法，所有要写基类方法使用，首先在插件中创建一个基类

将插件文件继承基类

思路是创建基类使用handler.cmd ,命令去获取系统信息，然后进行判断，然后去执行磁盘，cpu，网卡，内存等信息的收集；

基类代码：

class BasePlugin:

    def get_os(self,handler,hostname):

        os = handler.cmd("查询操作系统的命令",hostname)

        # return os

        return 'win32'

    def process(self,handler,hostname):

        os = self.get_os(handler,hostname)

        if os == 'win32': #测试判断执行win32

            return self.win(handler,hostname)

        else:

            return self.linux(handler,hostname)

    def win(self,handler,hostname):

        #约束派生类必须实现win方法

        raise NotImplementedError('handler() must Implemented.')

    def linux(self,handler,hostname):

        #约束派生类必须实现Linux方法

        raise NotImplementedError('handler() must Implemented.')

disk.py ，cpu.py，memory.py,network.py 代码;

from .base import BasePlugin

class Disk(BasePlugin):

    def win(self,handler,hostname):

        '''

        执行命令拿到结果磁盘

        :return:

        '''

        print("执行win方法")

        ret = handler.cmd('wmic diskdrive',hostname)[0:10]

        return ret

    def linux(self,handler,hostname):

        '''

        执行命令拿到结果磁盘

        :return:

        '''

        print("执行Linux方法")

        ret = handler.cmd('df -h',hostname)[0:10]

        return ret

from .base import BasePlugin

class Memory(BasePlugin):

    def win(self,handler,hostname):

        '''

        执行命令拿到结果-内存

        :return:

        '''

        print("执行win方法")

        ret = handler.cmd('wmic memphysical list brief',hostname)[0:10]

        return ret

    def linux(self,handler,hostname):

        '''

        执行命令拿到结果-内存

        :return:

        '''

        print("执行Linux方法")

        ret = handler.cmd('free',hostname)[0:10]

        return ret

from .base import BasePlugin

class CPU(BasePlugin):

    def win(self,handler,hostname):

        '''

        执行命令拿到结果-cpu

        :return:

        '''

        print("执行win方法")

        ret = handler.cmd('wmic cpu',hostname)[0:10]

        return ret

    def linux(self,handler,hostname):

        '''

        执行命令拿到结果-cpu

        :return:

        '''

        print("执行Linux方法")

        ret = handler.cmd('wmic cpu',hostname)[0:10]

        return ret

from .base import BasePlugin

class Network(BasePlugin):

    def win(self,handler,hostname):

        '''

        执行命令拿到结果-网卡

        :return:

        '''

        print("执行win方法")

        ret = handler.cmd('ipconfig',hostname)[0:10]

        return ret

    def linux(self,handler,hostname):

        '''

        执行命令拿到结果-网卡

        :return:

        '''

        print("执行Linux方法")

        ret = handler.cmd('ifconfig',hostname)[0:10]

        return ret

最后测试执行结果

巴特西

CMDB学习之三数据采集

最新文章

热门文章