1 # client.py
2 # ————————01CMDB获取服务器基本信息————————
3 from src import plugins #__init__.py
4 from lib.serialize import Json #转成字符串或者模式
5
6 # ————————02CMDB将服务器基本信息提交到API接口————————
7 import requests #伪造页面访问
8 from config import settings #文件配置
9 # ————————02CMDB将服务器基本信息提交到API接口————————
10
11 # ————————03CMDB信息安全API接口交互认证————————
12 import hashlib
13 import time
14 # ————————03CMDB信息安全API接口交互认证————————
15
16 # ————————04CMDB本地(Agent)模式客户端唯一标识(ID)————————
17 import os #操作系统层面执行
18 import shutil #高级的 文件、文件夹、压缩包 处理模块
19 # ————————04CMDB本地(Agent)模式客户端唯一标识(ID)————————
20 class AutoBase(object):
21
22 # ————————02CMDB将服务器基本信息提交到API接口————————
23 def __init__(self):
24 self.asset_api = settings.ASSET_API #ASSET_API = "http://127.0.0.1:8000/api/asset"
25
26 # ————————03CMDB信息安全API接口交互认证————————
27 self.key = settings.KEY # 用于API认证的KEY#KEY = '299095cc-1330-11e5-b06a-a45e60bec08b'
28 self.key_name = settings.AUTH_KEY_NAME # 'auth-key' API认证的请求头
29 # ————————03CMDB信息安全API接口交互认证————————
30
31 # ————————03CMDB信息安全API接口交互认证————————
32 def auth_key(self):#API接口认证
33 ha = hashlib.md5(self.key.encode('utf-8'))#认证的密码
34 time_span = time.time() #现在的时间戳 #1529819687.8867188
35 ha.update(bytes("%s|%f" % (self.key, time_span), encoding='utf-8'))#更新认证密码#密码+时间戳
36 encryption = ha.hexdigest() # 对‘时间密码’进行哈希
37 result = "%s|%f" % (encryption, time_span) #把‘时间密码’和 时间戳（解密用） 作为 API认证的请求头
38 print('‘时间密码’和 时间戳:',result)
39 return {self.key_name: result} # 'auth-key' API认证的请求头
40 # ————————03CMDB信息安全API接口交互认证————————
41
42 def post_asset(self, msg):#post方式向API接口提交资产信息
43 status = True#是否获取到信息
44 try:
45 # ————————03CMDB信息安全API接口交互认证————————
46 headers = {}
47 headers.update(self.auth_key())##认证的密码
48 # ————————03CMDB信息安全API接口交互认证————————
49 response = requests.post(
50 url=self.asset_api,
51 # ————————03CMDB信息安全API接口交互认证————————
52 headers=headers,
53 # ————————03CMDB信息安全API接口交互认证————————
54 json=msg
55 )
56 except Exception as e:
57 response = e
58 status = False #获取信息时出现错误
59 print(response.json())
60 # ————————02CMDB将服务器基本信息提交到API接口————————
61
62 def process(self):#派生类需要继承此方法，用于处理请求的入口
63 raise NotImplementedError('您必须实现过程的方法')
64
65 class AutoAgent(AutoBase):
66
67 # ————————04CMDB本地(Agent)模式客户端唯一标识(ID)————————
68 def __init__(self):
69 self.cert_file_path = settings.CERT_FILE_PATH# Agent模式保存服务器唯一ID的文件
70 super(AutoAgent, self).__init__()
71 """
72 super() 函数是用于调用父类(超类)的一个方法。
73 super 是用来解决多重继承问题的，直接用类名调用父类方法在使用单继承的时候没问题，
74 但是如果使用多继承，会涉及到查找顺序（MRO）、重复调用（钻石继承）等种种问题。
75 MRO 就是类的方法解析顺序表, 其实也就是继承父类方法时的顺序表。
76 """
77 def load_local_cert(self):# 获取本地唯一标识
78 if not os.path.exists(self.cert_file_path):#使用os.path.exists()方法可以直接判断文件(或者文件夹)是否存在
79 return None #如果文件(或者文件夹)不存在 #返回空
80 if os.path.isfile(self.cert_file_path): #判断是不是文件
81 with open(self.cert_file_path, mode='r') as f: #如果文件存在 #打开读取文件
82 data = f.read()#获取唯一ID的文件#read() 方法用于从文件读取指定的字节数，如果未给定或为负则读取所有。
83 if not data: # 如果文件没有内容 #返回空
84 return None
85 cert = data.strip() # strip() 方法用于移除字符串头尾指定的字符（默认为空格或换行符）或字符序列。
86 return cert # 返回 本地唯一标识
87 else: #防止有文件夹叫cert，导致无法创建cert文件
88 shutil.rmtree(self.cert_file_path)#就 递归的去删除文件夹 #这个目录以及子文件全部删除
89 return None #返回空 #去创建文件
90
91 def write_local_cert(self, cert):#写入本地唯一标识
92 with open(self.cert_file_path, mode='w') as f: #循环打开文件 #没有就创建 然后打开
93 f.write(cert) #写入本地唯一标识
94 # ————————04CMDB本地(Agent)模式客户端唯一标识(ID)————————
95
96 def process(self):
97 server_info = plugins.get_server_info()#获取本地基本信息
98
99 # ————————04CMDB本地(Agent)模式客户端唯一标识(ID)————————
100 if not server_info.status: #获取的基本信息是否出错
101 return
102 local_cert = self.load_local_cert() # 去获取本地唯一标识
103 if local_cert: #如果 获取本地唯一标识 后 #不是 return None
104 if local_cert == server_info.data['hostname']: #判断本地文件的唯一标识 是不是等于 获取的基本信息里主机名
105 pass #一样就什么都不干
106 else: #不一样就把 获取的基本信息里主机名 替换成 本地文件的唯一标识
107 #防止 恶意篡改 主机名 ，上报API时 错认为是 新资产
108 server_info.data['hostname'] = local_cert # 更新字典的主机名
109 else: # 获取本地唯一标识为None时 #把获取到的主机名写入本地做为唯一标识
110 self.write_local_cert(server_info.data['hostname'])
111 # ————————04CMDB本地(Agent)模式客户端唯一标识(ID)————————
112
113 server_json = Json.dumps(server_info.data)#json.dumps将 Python 对象编码成 JSON 字符串
114 print('提交资产信息:',server_json)
115 # ————————01CMDB获取服务器基本信息————————
116
117 # ————————02CMDB将服务器基本信息提交到API接口————————
118 self.post_asset(server_json)# post方式向接口提交资产信息
119 # ————————02CMDB将服务器基本信息提交到API接口————————
浙公网安备 33010602011771号