cmdb 可插拔式
实现三套方案,采集IP信息
首先应该能想到的代码:
if self.mode == 'agent': import subprocess res = subprocess.getoutput(cmd) return res elif self.mode == 'ssh': import paramiko # 创建SSH对象 ssh = paramiko.SSHClient() # 允许连接不在know_hosts文件中的主机 ssh.set_missing_host_key_policy(paramiko. AutoAddPolicy()) # 连接服务器 ssh.connect(hostname=self.hostname, port=22, username='root', password='root') # 执行命令 stdin, stdout, stderr = ssh.exec_command(cmd) # 获取命令结果 result = stdout.read() # 关闭连接 ssh.close() return result elif self.mode == 'salt': import salt.client local = salt.client.LocalClient() result = local.cmd(self.hostname, 'cmd.run', [cmd]) return result else: raise Exception('当前模式只支持 agent/ssh/salt !!!')
存在的问题很明显:
1、复用性差,需要将其封装成方法,然后在进行调用
2、高内聚低耦合原则(这一块代码是负责干啥的,其所有的代码都应该和这个功能是相关的)
举个栗子:
在函数中是: def getUserInfo(): #此函数体内,所有的代码,都应该和获取用户信息的这个目标功能相关联 ## 非 得要写处理订单相关的功能 在类中,也是一样的概念: class UserInfo(): def getUserInfo(): pass def login(): pass def logOut(): pass
采用高内聚低耦合的原则,迭代上述代码:
思路:
1.将硬盘或者cpu等代码封装成一个文件,此文件中涉及到的代码和这个文件的具体功能是相关的
采取仿Django的可插拔式采集:
a. conf.py 中的配置插件:
import os BASEDIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) USER = 'root' MODE = 'agent' ## ssh /salt DEBUG = True ### True 代表调试模式 False 代表上线正式采集 PLUGINS_DICT = { 'basic' : 'src.plugins.basic.Basic', 'board' : 'src.plugins.board.Board', 'cpu' : 'src.plugins.cpu.Cpu', 'disk' : 'src.plugins.disk.Disk', 'memory': 'src.plugins.memory.Memory', 'nic' : 'src.plugins.nic.Nic', }
b. src下的plugins目录 创建__init__.py , 此文件中有一个类:
#### 用来管理采集的插件的 import traceback #这个模块是用来查看详情错误信息的,不然用了异常捕获,获取的信息只有一点点 from lib.config.settings import settings #配置文件(高级配置和自定义都有) import importlib #关键模块,用来获取每个文件里面的类,然后调用Process方法。 class Plugins_Dict(): def __init__(self, hostname=None): self.mode = settings.MODE self.plugins_dict = settings.PLUGINS_DICT self.debug = settings.DEBUG self.hostname = hostname #### 从配置文件中读取插件配置,并执行相关的函数获取数据 def execute(self): ### 1.从配置文件中读取插件配置 res = {} for k, v in self.plugins_dict.items(): ### 'basic' : 'src.plugins.basic.Basic', ### k: basic ### v: 'src.plugins.basic.Basic' response = {'status':None, 'data':None} #### 2. 导入模块, 获取类, 并实例化 try: # 这里解压缩,一个是src.plugins.basic,一个是Basic module_path, class_name = v.rsplit('.', 1) ### 3. 加载模块,给一个路径,就能把名称空间加载过来 m = importlib.import_module(module_path) #### 4. 加载类, 通过反射,就可以在这个上面加载过来的名称空间 中拿到类。也就是栗子里的Basic类 cls = getattr(m, class_name) #### 5. 执行每一个插件类下面的process方法,并且传入两个参 数,第一个是函数对象,第二个是debug。因为这里的每个process代表的只是设备,你需要获取信息的设备,如网卡,磁盘等等,以什么样的方式采集信息才是重点,也就是这个函数对象所做的。debug就是判断是否是调试。 ret = cls().process(self.command_func, self.debug ) response['status'] = 10000 response['data'] = ret except Exception as e: response['status'] = 10001 #本来被捕获的异常打印出来只有一点点错误信息,无法精准判断错误位置和错误类型,用这个模块就可以知道了。 response['data'] = traceback.format_exc() res[k] = response return res #这个函数就是用来传进process方法里的。之前也写过这个方法了。放在了这个__init__文件里面。 def command_func(self, cmd): if self.mode == 'agent': import subprocess res = subprocess.getoutput(cmd) return res elif self.mode == 'ssh': import paramiko # 创建SSH对象 ssh = paramiko.SSHClient() # 允许连接不在know_hosts文件中的主机 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy()) # 连接服务器 ssh.connect(hostname=self.hostname, port=22, username='root', password='root') # 执行命令 stdin, stdout, stderr = ssh.exec_command(cmd) # 获取命令结果 result = stdout.read() # 关闭连接 ssh.close() return result elif self.mode == 'salt': import salt.client local = salt.client.LocalClient() result = local.cmd(self.hostname, 'cmd.run', [cmd]) return result else: raise Exception('当前模式只支持 agent/ssh/salt !!!')
这些都是插件,就是一个个设备,随便拿一个看一下,就拿cpu来看
import os from lib.config.settings import settings class Cpu(object): def __init__(self): pass @classmethod def initial(cls): return cls() def process(self, command_func, debug): #如果是debug模式,就返回一些已经有了的准备好了的信息。 if debug: output = open(os.path.join(settings.BASEDIR, 'files /cpuinfo.out'), 'r',encoding='utf8').read() else: output = command_func("cat /proc/cpuinfo") return self.parse(output) #这个函数就是用来解析拿到的结果的。 def parse(self, content): """ 解析shell命令返回结果 :param content: shell 命令结果 :return:解析后的结果 """ response = {'cpu_count': 0, 'cpu_physical_count': 0, 'cpu_model': ''} cpu_physical_set = set() content = content.strip() for item in content.split('\n\n'): for row_line in item.split('\n'): key, value = row_line.split(':') key = key.strip() if key == 'processor': response['cpu_count'] += 1 elif key == 'physical id': cpu_physical_set.add(value) elif key == 'model name': if not response['cpu_model']: response['cpu_model'] = value response['cpu_physical_count'] = len(cpu_physical_set) return response
这是一个Cpu类, 在__init__里面,通过反射拿到了这个类,并实例化,再执行了process方法。
相关推荐
projava 2020-11-14
WanKaShing 2020-11-12
airfish000 2020-09-11
tryfind 2020-09-14
yegen00 2020-09-10
kkaazz 2020-09-03
风语者 2020-09-02
BraveWangDev 2020-08-19
lichuanlong00 2020-08-15
gsl 2020-08-15
pandaphinex 2020-08-09
yhuihon 2020-08-09
CheNorton 2020-08-02
xiangqiao 2020-07-28
hpujsj 2020-07-26
hpujsj 2020-07-26
sshong 2020-07-19
BraveWangDev 2020-07-19
annan 2020-07-18