实现三套方案，采集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方法。