CMDB项目要点总结之中控机

1.基于paramiko对远程主机执行命令操作

秘钥形式



private_key = paramiko.RSAKey.from_private_key_file('c:/Users/用户名/.ssh/id_rsa')

# 创建ssh对象

ssh = paramiko.SSHClient()

# 运行连接不在know_host文件中的主机

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

# 连接服务器

ssh.connect(hostname='主机名', port='端口号', username='用户名', pkey=private_key)

# 执行命令

stdin, stdout, stderr = ssh.exec_command("命令")

# 获取结果

result = stdout.read()

# 关闭连接

ssh.close()

2.线程池提高效率

from concurrent.futures import ThreadPoolExecutor

# 创建一个十个线程的线程池

def task(i):

   print("我是任务%s" % i)

pool = ThreadPoolExecutor(10)

for i in range(10):

   pool.submit(task,i)

3.基于logging模块进行日志的提取

import logging

import traceback

# import settings

class Logger(object):

    def __init__(self, file_path, level):

        file_handler = logging.FileHandler(file_path, 'a', encoding='utf-8')

        fmt = logging.Formatter(fmt="%(asctime)s - %(name)s - %(levelname)s: %(message)s")

        file_handler.setFormatter(fmt)

        self.logger = logging.Logger('cmdb', level=level)

        self.logger.addHandler(file_handler)

    def error(self, msg):

        self.logger.error(msg)

logger = Logger("你的日志存放路径", logging.DEBUG)

logger.error(traceback.format_exc())  # 获取异常堆栈信息

4.基于类的封装对数据进行封装

class BaseResponse(object):

    def __init__(self, status=True, data=None, error=None):

        self.status = status

        self.data = data

        self.error = error

    @property  # 作用：让调用着不用加()

    def dict(self):

        return self.__dict__

5.基于工厂模式和反射对资产数据进行获取

def get_server_info(ssh, hostname):

	server_info = {}

	for key, path in settings.PLUGIN_CLASS_DICT.items():

		module_path, class_name = path.rsplit('.', maxsplit=1)

		try:

			module = importlib.import_module(module_path)

		except Exception as e:

			print(e)

		cls = getattr(module, class_name)

		plugin_obj = cls()

		info = plugin_obj.process(ssh, hostname)

		server_info[key] = info

	return server_info

# settings.py

# 获取计算机信息配置

PLUGIN_CLASS_DICT = {

	'basic': 'lib.plugin.basic.BasicPlugin',

	'disk': 'lib.plugin.disk.DiskPlugin',

	'memory': 'lib.plugin.memory.MemoryPlugin',

	'network': 'lib.plugin.network.NetworkPlugin',

}

# disk.py

import traceback

from lib.utils.log import logger

from lib.utils.response import BaseResponse

class DiskPlugin(BasePlugin):

	"""

	获取磁盘信息

	"""

	def process(self, ssh, hostname):

		result = BaseResponse()

		try:

			# output = ssh(hostname, 'MegaCli -PDList -aALL')

			output = open('files/disk.out').read()

			result.data = output

			output.close()

		except Exception as e:

			logger.error(traceback.format_exc())

			result.status = False

			result.error = traceback.format_exc()

		return result.dict

6.采集资产用到的命令

cpu:

	cat /proc/cpuinfo

主板:

	dmidecode -t1

内存:

	dmidecode -q -t 17 2>/dev/null

网卡(salt源码):

	ip link show

	ip addr show

硬盘：

	MegaCli -PDList -aALL

巴特西

CMDB项目要点总结之中控机

1.基于paramiko对远程主机执行命令操作

秘钥形式

2.线程池提高效率

3.基于logging模块进行日志的提取

4.基于类的封装对数据进行封装

5.基于工厂模式和反射对资产数据进行获取

6.采集资产用到的命令

最新文章

热门文章