CMDB学习之三数据采集

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

将插件文件继承基类

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

基类代码：

class BasePlugin:def get_os(self,handler,hostname):os = handler.cmd("查询操作系统的命令",hostname)# return osreturn 'win32'def process(self,handler,hostname):os = self.get_os(handler,hostname)if os == 'win32': #测试判断执行win32return 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 BasePluginclass Disk(BasePlugin):def win(self,handler,hostname):'''执行命令拿到结果磁盘:return:'''print("执行win方法")ret = handler.cmd('wmic diskdrive',hostname)[0:10]return retdef linux(self,handler,hostname):'''执行命令拿到结果磁盘:return:'''print("执行Linux方法")ret = handler.cmd('df -h',hostname)[0:10]return ret

from .base import BasePluginclass Memory(BasePlugin):def win(self,handler,hostname):'''执行命令拿到结果-内存:return:'''print("执行win方法")ret = handler.cmd('wmic memphysical list brief',hostname)[0:10]return retdef 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 retdef 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 retdef linux(self,handler,hostname):'''执行命令拿到结果-网卡:return:'''print("执行Linux方法")ret = handler.cmd('ifconfig',hostname)[0:10]return ret