python cmdb资产_CMDB资产采集

importosimportjsonimporttimeimporthashlibimportrequestsfrom src importpluginsfrom lib.serialize importJsonfrom lib.log importLoggerfrom config importsettingsfrom concurrent.futures importThreadPoolExecutorclassAutoBase(object):def __init__(self):

self.asset_api=settings.ASSET_API

self.key=settings.KEY

self.key_name=settings.AUTH_KEY_NAMEdefauth_key(self):"""接口认证

:return:"""ha= hashlib.md5(self.key.encode('utf-8'))

time_span=time.time()

ha.update(bytes("%s|%f" % (self.key, time_span), encoding='utf-8'))

encryption=ha.hexdigest()

result= "%s|%f" %(encryption, time_span)return{self.key_name: result}defget_asset(self):"""get方式向获取未采集的资产

:return: {"data": [{"hostname": "c1.com"}, {"hostname": "c2.com"}], "error": null, "message": null, "status": true}"""

try:

headers={}

headers.update(self.auth_key())

response=requests.get(

url=self.asset_api,

headers=headers

)exceptException as e:

response=ereturnresponse.json()def post_asset(self, msg, callback=None):"""post方式向接口提交资产信息

:param msg:

:param callback:

:return:"""status=Truetry:

headers={}

headers.update(self.auth_key())

response=requests.post(

url=self.asset_api,

headers=headers,

json=msg

)exceptException as e:

response=e

status=Falseifcallback:

callback(status, response)defprocess(self):"""派生类需要继承此方法，用于处理请求的入口

:return:"""

raise NotImplementedError('you must implement process method')defcallback(self, status, response):"""提交资产后的回调函数

:param status: 是否请求成功

:param response: 请求成功，则是响应内容对象；请求错误，则是异常对象

:return:"""

if notstatus:

Logger().log(str(response), False)returnret=json.loads(response.text)if ret['code'] == 1000:

Logger().log(ret['message'], True)else:

Logger().log(ret['message'], False)classAutoAgent(AutoBase):def __init__(self):

self.cert_file_path=settings.CERT_FILE_PATH

super(AutoAgent, self).__init__()defload_local_cert(self):"""获取本地以为标识

:return:"""

if notos.path.exists(self.cert_file_path):returnNone

with open(self.cert_file_path, mode='r') as f:

data=f.read()if notdata:returnNone

cert=data.strip()returncertdefwrite_local_cert(self, cert):"""写入本地以为标识

:param cert:

:return:"""

if notos.path.exists(self.cert_file_path):

os.makedirs(os.path.basename(self.cert_file_path))

with open(settings.CERT_FILE_PATH, mode='w') as f:

f.write(cert)defprocess(self):"""获取当前资产信息

1. 在资产中获取主机名 cert_new

2. 在本地cert文件中获取主机名 cert_old

如果cert文件中为空，表示是新资产

- 则将 cert_new 写入该文件中，发送数据到服务器(新资产)

如果两个名称不相等

- 如果 db=new 则，表示应该主动修改，new为唯一ID

- 如果 db=old 则，表示

:return:"""server_info=plugins.get_server_info()if notserver_info.status:returnlocal_cert=self.load_local_cert()iflocal_cert:if local_cert == server_info.data['hostname']:pass

else:

server_info.data['hostname'] =local_certelse:

self.write_local_cert(server_info.data['hostname'])

server_json=Json.dumps(server_info.data)

self.post_asset(server_json, self.callback)classAutoSSH(AutoBase):defprocess(self):"""根据主机名获取资产信息，将其发送到API

:return:"""task=self.get_asset()if not task['status']:

Logger().log(task['message'], False)

pool= ThreadPoolExecutor(10)for item in task['data']:

hostname= item['hostname']

pool.submit(self.run, hostname)

pool.shutdown(wait=True)defrun(self, hostname):

server_info=plugins.get_server_info(hostname)

server_json=Json.dumps(server_info.data)

self.post_asset(server_json, self.callback)classAutoSalt(AutoBase):defprocess(self):"""根据主机名获取资产信息，将其发送到API

:return:

{

"data": [ {"hostname": "c1.com"}, {"hostname": "c2.com"}],

"error": null,

"message": null,

"status": true

}"""task=self.get_asset()if not task['status']:

Logger().log(task['message'], False)#创建线程池：最大可用线程10

pool = ThreadPoolExecutor(10)#"data": [ {"hostname": "c1.com"}, {"hostname": "c2.com"}],

for item in task['data']:#c1.com c2.com

hostname = item['hostname']

pool.submit(self.run, hostname)#run(c1.com) 1

#run(c2.com) 2

pool.shutdown(wait=True)defrun(self, hostname):#获取指定主机名的资产信息

#{'status': True, 'message': None, 'error': None, 'data': {'disk': , 'main_board': , 'nic': , 'memory': , 'os_platform': 'linux', 'os_version': 'CentOS release 6.6 (Final)', 'hostname': 'c1.com', 'cpu': }}

server_info =plugins.get_server_info(hostname)#序列化成字符串

server_json =Json.dumps(server_info.data)#发送到API

self.post_asset(server_json, self.callback)