CMDB资产采集笔记

作者：天王2502871933 | 来源：互联网 | 2023-06-05 09:49

一、资产采集四种方式1.Agent方式API：Django接收数据并入库程序：放置在每台服务器应用场景：针对服务器较多的公司步骤一:#执行

一、资产采集四种方式

1. Agent方式

API&＃xff1a;Django接收数据并入库程序&＃xff1a;放置在每台服务器应用场景&＃xff1a;针对服务器较多的公司步骤一:#执行本地命令的库import subprocesssub &＃61; subprocess.getoutput("要执行的命令名")每台机器通过用户名密码链接数据库&＃xff0c;获取要执行的命令步骤二: 采集数据import subprocess# 采集到本机运行ipconfig命令的输出结果result &＃61; subprocess.getoutput("ipconfig")print(result)步骤三: 筛选整理数据# 正则等方式result_dic &＃61; {"网络信息":result,}步骤四:发送数据到api# 以post方式将数据发送到urlimport requestsrequests.post("url", result_dic)

2. SSH方式

import paramiko # 通过paramiko库链接各台机器&＃xff0c;获取命令 #缺点是慢API&＃xff1a;Django接收数据并入库程序&＃xff1a;放在中控机应用场景&＃xff1a;针对服务器较少的公司SSH方式有三种工具&＃xff0c;paramiko、Fabric、Ansible&＃xff0c;其中Fabric、Ansible内部原理也是通过paramiko来实现得。 paramiko安装&＃xff1a;pip3 install paramikoandible,fabic等工具对paramiko进行了封装&＃xff0c;方便使用速度更快一些&＃xff01;import requests,paramiko # 获取今天未采集的主机名 # 从url中获取信息 host_list &＃61; requests.get("url")# 通过paramiko连接远程服务器执行命令 # 创建ssh对象 ssh &＃61; paramiko.SSHClient()# 允许连接不在know_hosts文件中的主机 ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())# 连接服务器 ssh.connect(hostname&＃61;"hostname", port&＃61;22, username&＃61;"root", password&＃61;"password")# 执行命令 stdin, stdout, stderr &＃61; ssh.exec_command("hostname")# 获取结果 result &＃61; stdout.read() print(result)# 关闭连接 ssh.close()# 筛选整理数据 result_dic &＃61; {"网络信息":result, }# 发送获取的数据 requests.post("url", result_dic)

3. saltstack【python语言】

方式同ssh&＃xff0c;原理不同优点&＃xff1a;提高速度&＃xff0c;开发成本低缺点&＃xff1a;依赖saltstack工具可以远程执行命令http://www.cnblogs.com/wupeiqi/articles/6415436.htmlAPI&＃xff1a;Django接收数据并入库应用场景&＃xff1a;针对服务器较多的公司【推荐】master:v &＃61; subprocess.getoutput(&＃39;salt "*" cmd.run "ls"&＃39;)saltstack内部原理是通过RPC来实现得消息队列步骤一: 安装saltstackrpm --import https://repo.saltstack.com/yum/redhat/7/x86_64/latest/SALTSTACK-GPG-KEY.pubyum install salt-master 服务端yum install salt-minion 客户端步骤二: Master准备a.配置文件&＃xff0c;修改监听ip/etc/salt/masterinterface: 本机ip地址b.启动Master/etc/init.d/salt-master start步骤三: Slave准备a.修改配置文件&＃xff0c;连接哪个master/etc/salt/minionmaster: 远程master的ipb.启动Slave/etc/init.d/salt-slave start步骤四: 创建关系Master&＃xff1a;salt-key -L 查看哪些连接salt-key -a 接受连接salt-key -A 接收全部连接步骤五: 执行命令Master&＃xff1a;salt "连接" cmd.run "命令"# salt 是一个.py文件 #!/usr/bin/python # Publish commands to the salt system from the command line on the master.from salt.scripts import salt_main if __name__ &＃61;&＃61; &＃39;__main__&＃39;:salt_main()# 在Python文件中使用salt的方法&＃xff08;需要提前配置好&＃xff09; import salt.client local &＃61; salt.client.LocalClient() result &＃61; local.cmd(&＃39;对方ip&＃39;, &＃39;cmd.run&＃39;, [&＃39;命令&＃39;]) # result是一个字典 result.keys() # 命令 result.values() # 执行结果

4. puppet【ruby语言】

内部原理&＃xff1a;puppet客户端每30秒钟自动汇报数据给puppet服务端。应用场景&＃xff1a;主要是针对已经在使用puppet的公司。基于ruby 优点&＃xff1a;自动汇报缺点&＃xff1a;要用ruby写

二、高级配置文件路径

1.目录结构

2.整合配置文件的代码

import os import importlib from . import global_settingsclass Settings(object):def __init__(self):# ------------ 找到默认配置文件----------------for name in dir(global_settings):if name.upper(): # 配置文件里的变量名需要大写value &＃61; getattr(global_settings,name) # 拿到值setattr(self,name,value) # 设置值# ------------ 找到用户自定义配置文件------------settings_module &＃61; os.environ.get(&＃39;USER_SETTINGS&＃39;) # 加载设置到环境变量中的配置文件路径名if not settings_module: # 如果用户没有自定义配置文件或没有初始化自定义文件return # 直接返回不做其他操作user_settings &＃61; importlib.import_module(settings_module)for name in dir(user_settings):if name.upper(): # 配置文件里的变量名需要大写value &＃61; getattr(user_settings,name) # 拿到值setattr(self,name,value) # 设置值settings &＃61; Settings() # 实例化对象

3.加载自定义配置文件的名称到环境变量中

# 以下代码放置需要执行的py文件中 import os os.environ[&＃39;USER_SETTINGS&＃39;] &＃61; "config.settings" from lib.conf.config import settings # 通过settings可以点出在自定义或默认配置文件中的相关变量名

三、可插拔式插件 &＃xff08;资产采集&＃xff09;

1.目录结构

2.流程介绍

主程序start.py 去执行 script.run()

import os os.environ[&＃39;USER_SETTINGS&＃39;] &＃61; "config.settings" # 初始化 import sys BASEDIR &＃61; os.path.dirname(os.path.dirname(os.path.abspath(__file__))) sys.path.append(BASEDIR) #上级目录加入环境变量中 from src import script # 导入执行代码 if __name__ &＃61;&＃61; &＃39;__main__&＃39;:script.run()

script.py中根据MODE去执行相应的方法

MODE是资产采集中四种方法中的一种

from lib.conf.config import settings from .client import Agent from .client import SSHSALTdef run():if settings.MODE &＃61;&＃61; &＃39;AGENT&＃39;: # 配置文件中MODE的值为AGENT 或 ALT,SSHobj &＃61; Agent()else:obj &＃61; SSHSALT()obj.execute()

client.py中主要是拿到查到的主机的硬盘、CPU、内存等信息发给API

如果是Agent&＃xff1a;其中Agent 每个客户端自己查询到信息后直接发给API

如果是SSHSALT&＃xff1a;中控机服务器先向API发送请求得到要查询的主机列表

然后遍历列表&＃xff0c;

PluginManager(host).exec_plugin() # 依次传入每个主机的host信息,并拿到返回的查询信息

import requests import json from src.plugins import PluginManager from lib.conf.config import settings# 基类只有向API发送查询到信息的方法 class Base(object):def post_asset(self,server_info):requests.post(settings.API,json&＃61;server_info)# body: json.dumps(server_info) 直接写json&＃61;server_info可以把里面的东西改成json字符串# headers&＃61; {&＃39;content-type&＃39;:&＃39;application/json&＃39;}# request.body# json.loads(request.body)# 继承基类,每个客户端自己向API发送查询到的信息 class Agent(Base):def execute(self):server_info &＃61; PluginManager().exec_plugin()self.post_asset(server_info) # 向API发送查询到信息class SSHSALT(Base):def get_host(self):# 获取未采集信息的主机列表response &＃61; requests.get(settings.API) # 中控机向API发送get请求获取要查询的主机信息result &＃61; json.loads(response.text) # "{status:&＃39;True&＃39;,data: [&＃39;c1.com&＃39;,&＃39;c2.com&＃39;]}"if not result[&＃39;status&＃39;]: # 状态发生异常,则直接结束跳过异常的主机returnreturn result[&＃39;data&＃39;]def execute(self):host_list &＃61; self.get_host() # 拿到主机列表清单for host in host_list: # 遍历拿到每个主机信息server_info &＃61; PluginManager(host).exec_plugin() # 依次传入每个主机的host信息,并拿到返回的查询信息self.post_asset(server_info) # 向API发送查询到信息

在plugins目录下的__init__.py文件中

函数 exec_plugin 获取所有的插件&＃xff0c;执行插件里边的方法并获取插件返回值

从配置文件中获取到插件路径和类名

model_path,class_name &＃61; v.rsplit(&＃39;.&＃39;,1) # 从右向左按点分割一次&＃xff0c;并赋值给相应变量

# &＃39;basic&＃39;: "src.plugins.basic.Basic" 数据结构 v 是"src.plugins.basic.Basic"

model_path,class_name &＃61; v.rsplit(&＃39;.&＃39;,1) # 从右向左按点分割一次&＃xff0c;并赋值给相应变量 model_file &＃61; importlib.import_module(model_path) # 以字符串的形式导入模块 cls &＃61; getattr(model_file,class_name) # 利用反射机制拿到类对象

import importlib import traceback from lib.conf.config import settingsclass PluginManager(object):def __init__(self,hostname&＃61;None):self.hostname &＃61; hostnameself.plugin_dict &＃61; settings.PLUGINS_DICT # 插件存放的目录self.mode &＃61; settings.MODEself.debug &＃61; settings.DEBUGif self.mode &＃61;&＃61; &＃39;SSH&＃39;:# 拿到配置文件中用来SSH登录的相关信息self.ssh_user &＃61; settings.SSH_USERself.ssh_pwd &＃61; settings.SSH_PWDself.ssh_port &＃61; settings.SSH_PORTself.ssh_key &＃61; settings.SSH_KEYdef exec_plugin(self):"""获取所有的插件&＃xff0c;并执行获取插件返回值:return:"""response &＃61; {}for k,v in self.plugin_dict.items():# &＃39;basic&＃39;: "src.plugins.basic.Basic" 数据结构ret &＃61; {&＃39;status&＃39;:True,&＃39;data&＃39;:None}try:model_path,class_name &＃61; v.rsplit(&＃39;.&＃39;,1) # 从右向左按点分割一次&＃xff0c;并赋值给相应变量model_file &＃61; importlib.import_module(model_path) # 以字符串的形式导入模块cls &＃61; getattr(model_file,class_name) # 拿到类对象if hasattr(cls,&＃39;initial&＃39;): # 如果类中有initial方法obj &＃61; cls.initial() # 执行类方法initial 并把返回值赋给objelse:obj &＃61; cls() # 否则直接实例化类对象# obj.process(self.command,self.debug) 执行插件中的process方法&＃xff0c;传入command函数和debug信息result &＃61; obj.process(self.command,self.debug) # result &＃61; "根据v获取类&＃xff0c;并执行其方法采集资产"ret[&＃39;data&＃39;] &＃61; resultexcept Exception as e:ret[&＃39;status&＃39;] &＃61; False# traceback.format_exc() 可以返回更详细的错误信息字符串ret[&＃39;data&＃39;] &＃61; "[%s][%s] 采集数据出现错误 : %s" % (self.hostname if self.hostname else "AGENT", k, traceback.format_exc())response[k] &＃61; retreturn responsedef command(self,cmd):if self.mode &＃61;&＃61; "AGENT":return self.__agent(cmd)elif self.mode &＃61;&＃61; "SSH":return self.__ssh(cmd)elif self.mode &＃61;&＃61; "SALT":return self.__salt(cmd)else:raise Exception(&＃39;模式只能是 AGENT/SSH/SALT&＃39;)def __agent(self,cmd): # 如果配置文件中的方法是AGENTimport subprocessoutput &＃61; subprocess.getoutput(cmd)return outputdef __ssh(self,cmd): # 如果配置文件中的方法是SSHimport paramiko# 秘钥连接# private_key &＃61; paramiko.RSAKey.from_private_key_file(self.ssh_key)# ssh &＃61; paramiko.SSHClient()# ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())# ssh.connect(hostname&＃61;self.hostname, port&＃61;self.ssh_port, username&＃61;self.ssh_user, pkey&＃61;private_key)# stdin, stdout, stderr &＃61; ssh.exec_command(cmd)# result &＃61; stdout.read()# ssh.close()# 用户名密码连接ssh &＃61; paramiko.SSHClient()ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())ssh.connect(hostname&＃61;self.hostname,port&＃61;self.ssh_port,username&＃61;self.ssh_user,password&＃61;self.ssh_pwd)stdin,stdout,stderr &＃61; ssh.exec_command(cmd)result &＃61; stdout.read() # 读到返回的信息ssh.close()return resultdef __salt(self,cmd): # 也是通过操作subprocess# import salt.client# local &＃61; salt.client.LocalClient()# result &＃61; local.cmd(self.hostname, &＃39;cmd.run&＃39;, [cmd])# return result[self.hostname]salt_cmd &＃61; "salt &＃39;%s&＃39; cmd.run &＃39;%s&＃39;"%(self.hostname,cmd,)import subprocessoutput &＃61; subprocess.getoutput(salt_cmd)return output # 读到返回的信息

四、唯一标识和线程池

1.唯一标识&＃xff1a;主板SN&＃xff08;不准确&＃xff09;

标准化&＃xff1a;
- 主机名不重复&＃xff08;装机同时&＃xff0c;主机名在cmdb中设置&＃xff09;

步骤&＃xff1a;
a. 装系统&＃xff0c;初始化软件&＃xff08;CMDB&＃xff09;&＃xff0c;运行CMDB&＃xff1a;
- 通过命令获取主机名
- 写入本地指定文件
b. 将资产信息发送到API

c. 获取资产信息&＃xff1a;
- 本地文件主机名 !&＃61; 命令获取的主机名&＃xff08;按照文件中的主机名&＃xff09;
- 本地文件主机名 &＃61;&＃61; 命令获取的主机名

服务器资产采集&＃xff08;Agent&＃xff09;&＃xff1a;
a. 第一次&＃xff1a;文件不存在&＃xff0c;或内容为空&＃xff1b;
采集资产&＃xff1a;
- 主机名写入文件
- 发送API

b. 第N次&＃xff1a;采集资产&＃xff0c;主机名&＃xff1a;文件中获取
SSH或Salt&＃xff1a;
中控机&＃xff1a;获取未采集主机名列表&＃xff1a;【c1.com 】

# 继承基类,每个客户端自己向API发送查询到的信息
class Agent(Base):def execute(self):# 获取主机名唯一标识server_info &＃61; PluginManager().exec_plugin()hostname &＃61; server_info[&＃39;basic&＃39;][&＃39;data&＃39;][&＃39;hostname&＃39;]certname &＃61; open(settings.CERT_PATH, &＃39;r&＃39;, encoding&＃61;&＃39;utf-8&＃39;).read().strip() # 配置文件里拿到初始主机名if not certname: # 第一次的时候,把主机名写入文件with open(settings.CERT_PATH, &＃39;w&＃39;, encoding&＃61;&＃39;utf-8&＃39;) as f:f.write(hostname)else: # 以配置文件的主机名为准server_info[&＃39;basic&＃39;][&＃39;data&＃39;][&＃39;hostname&＃39;] &＃61; certnameself.post_asset(server_info) # 向API发送查询到信息

View Code

2.线程池

def run(self,host):server_info &＃61; PluginManager(host).exec_plugin() # 依次传入每个主机的host信息,并拿到返回的查询信息self.post_asset(server_info) # 向API发送查询到信息def execute(self):from concurrent.futures import ThreadPoolExecutorhost_list &＃61; self.get_host() # 拿到主机列表清单pool &＃61; ThreadPoolExecutor(10) # 创建的线程池中有10个线程for host in host_list: # 遍历拿到每个主机信息pool.submit(self.run,host) # 把run函数放入线程池

View Code

五、资产入库&＃xff08;API 资产入库同时对外提供数据访问接口&＃xff09;

1.后台API拿到发过来的主机信息

server_info &＃61; json.loads(request.body.decode(&＃39;utf-8&＃39;)) # 查询到的主机信息

　　 hostname &＃61; server_info[&＃39;basic&＃39;][&＃39;data&＃39;][&＃39;hostname&＃39;] # 拿到客户端自己查到的主机名

2.根据主机名查到数据库中未更新的老资产主机对象

server_obj &＃61; models.Server.objects.filter(hostname&＃61;hostname).first() # 服务器对象

3.查不到则直接退出

if not server_obj:

return HttpResponse(&＃39;当前主机名未在资产中录入&＃39;)

4.录入资产信息

# -------------- 处理硬盘、网卡、内存等的信息资产入库----------------
PluginManager(server_info, server_obj, hostname).exec_plugin()
我们可以利用python的反射机制来定制插件

PluginManager类写在__init__.py里
来看一下代码

import importlib
from autoserver import settingsclass PluginManager(object):def __init__(self, server_info,server_obj,hostname):self.server_info &＃61; server_infoself.server_obj &＃61; server_objself.hostname &＃61; hostnameself.plugin_dict &＃61; settings.PLUGINS_DICT #引入配置文件中的插件路径信息# &＃39;disk&＃39;: "api.service.disk.Disk"def exec_plugin(self):"""获取所有的插件&＃xff0c;并执行获取插件返回值:return:"""for k, v in self.plugin_dict.items():# &＃39;disk&＃39;: "api.service.disk.Disk", 数据结构ret &＃61; {&＃39;status&＃39;: True, &＃39;data&＃39;: None}try: # -------借鉴Django中间件的形式------------model_path, class_name &＃61; v.rsplit(&＃39;.&＃39;, 1) # 从右向左按点分割一次&＃xff0c;并赋值给相应变量model_file &＃61; importlib.import_module(model_path) # 以字符串的形式导入模块cls &＃61; getattr(model_file, class_name) # 拿到类对象if cls:if hasattr(cls, &＃39;initial&＃39;): # 如果类中有initial方法obj &＃61; cls.initial() # 执行类方法initial 并把返回值赋给objelse:obj &＃61; cls() # 否则直接实例化类对象obj.run(self.server_info,self.server_obj,self.hostname) # 资产入库"except Exception as e:passreturn

__init__.py

没什么亮点主要是 1.以字符串的形式导入模块&＃xff0c;2.python机制来获取类对象和方法
再来看一下disk.py里面的代码

from repository import modelsclass Disk(object):def __init__(self):pass&＃64;classmethoddef initial(cls):return cls()def run(self,server_info,server_obj,hostname):# 采集信息出错时if not server_info[&＃39;disk&＃39;][&＃39;status&＃39;]: # 采集硬盘信息时报错了的话models.ErrorLog.objects.create(content&＃61;server_obj[&＃39;disk&＃39;][&＃39;data&＃39;],asset_obj&＃61;server_obj.asset,title&＃61;&＃39;【%s】硬盘采集错误信息&＃39;%(hostname))# 新磁盘字典信息new_disk_dict &＃61; server_info[&＃39;disk&＃39;][&＃39;data&＃39;]"""slot 插槽位; pd_type 磁盘类型 ; capacity 磁盘容量GB; model 磁盘型号;{&＃39;5&＃39;: {&＃39;slot&＃39;: &＃39;5&＃39;, &＃39;pd_type&＃39;: &＃39;SAS&＃39;, &＃39;capacity&＃39;: &＃39;279.396&＃39;, &＃39;model&＃39;: &＃39;SEAGATE ST300MM0006 LS08S0K2B5NV&＃39;},&＃39;3&＃39;: {&＃39;slot&＃39;: &＃39;3&＃39;, &＃39;pd_type&＃39;: &＃39;SAS&＃39;, &＃39;capacity&＃39;: &＃39;279.396&＃39;, &＃39;model&＃39;: &＃39;SEAGATE ST300MM0006 LS08S0K2B5AH&＃39;},}"""# 旧磁盘列表信息old_disk_list &＃61; models.Disk.objects.filter(server_obj&＃61;server_obj)"""Queryset 列表字典类型&＃xff0c;多个硬盘对象[Disk(&＃39;slot&＃39;:5,capacity:476...)Disk(&＃39;slot&＃39;:4,capacity:476...)]"""new_slot_list &＃61; list(new_disk_dict.keys()) # 那新磁盘字典信息中的key 即槽位号old_slot_list &＃61; []for row in old_disk_list: # 遍历拿到数据库中旧的磁盘号
old_slot_list.append(row.slot)# 交集更新【5】则更新数据库的磁盘信息update_list &＃61; set(new_slot_list).intersection(old_slot_list) # 交集列表# 差集创建【3】则创建新的磁盘信息进数据库create_list &＃61; set(new_slot_list).difference(old_slot_list) # 差集列表# 差集删除【4】则把数据库里的不在返回来的新磁盘字典里的删除del_list &＃61; set(old_slot_list).difference(new_slot_list)if del_list:# 删除models.Disk.objects.filter(server_obj&＃61;server_obj,slot__in&＃61;del_list).delete()# 记录日志models.AssetRecord.objects.create(asset_obj&＃61;server_obj.asset,content&＃61;&＃39;移除硬盘%s&＃39;%(&＃39;、&＃39;.join(del_list)))# 新增record_list &＃61; []for slot in create_list:disk_dict &＃61; new_disk_dict[slot] # {&＃39;capacity&＃39;: &＃39;476.939&＃39;, &＃39;slot&＃39;: &＃39;4&＃39;, &＃39;model&＃39;: &＃39;S1AXNSAF303909M Samsung SSD 840 PRO Seriesdisk_dict[&＃39;server_obj&＃39;] &＃61; server_obj # 多增加一个字段models.Disk.objects.create(**disk_dict) # 以字典的方式增加tmp &＃61; &＃39;新增硬盘插槽位【{slot}】,磁盘类型【{pd_type}】,磁盘容量GB【{capacity}】,model磁盘型号【{model}】&＃39;.format(**disk_dict)record_list.append(tmp) # 依次把每次新增的信息加入一个列表中# 记录日志if record_list:con &＃61; &＃39;;&＃39;.join(record_list)models.AssetRecord.objects.create(asset_obj&＃61;server_obj.asset,content&＃61;con)# 更新cord_list &＃61; []row_map &＃61; {&＃39;capacity&＃39;: &＃39;容量&＃39;, &＃39;pd_type&＃39;: &＃39;类型&＃39;, &＃39;model&＃39;: &＃39;型号&＃39;}for slot in update_list:new_dick_row &＃61; new_disk_dict[slot]# {&＃39;slot&＃39;: &＃39;3&＃39;, &＃39;pd_type&＃39;: &＃39;SAS&＃39;, &＃39;capacity&＃39;: &＃39;279.396&＃39;, &＃39;model&＃39;: &＃39;SEAGATE ST300MM0006 LS08S0K2B5AH&＃39;}old_disk_row &＃61; models.Disk.objects.filter(slot&＃61;slot,server_obj&＃61;server_obj).first()for k,v in new_dick_row.items(): # 遍历新信息的键和值# k: capacity;slot;pd_type;model# v: &＃39;476.939&＃39;&＃39;xxies DXM05B0Q&＃39;&＃39;SATA&＃39;value &＃61; getattr(old_disk_row,k) # 拿到数据库中旧的值if v !&＃61; value: # 如果新值和旧值不一样&＃xff0c;则说明需要更新cord_list.append(&＃39;槽位%s %s由%s变更为%s&＃39;.format(slot,row_map[k],value,v))setattr(old_disk_row,k,v) # 更新数据库中的值old_disk_row.save() # 写入数据库# 记录日志if cord_list:tent &＃61; ";".join(cord_list)models.AssetRecord.objects.create(asset_obj&＃61;server_obj.asset,content&＃61;tent)

disk.py

在里边&＃xff1a;

新磁盘字典信息 new_disk_dict &＃61; server_info[&＃39;disk&＃39;][&＃39;data&＃39;]

旧磁盘列表信息 old_disk_list &＃61; models.Disk.objects.filter(server_obj&＃61;server_obj)

　　拿到新插槽号列表 new_slot_list &＃61; list(new_disk_dict.keys())

　　拿到旧插槽号列表 old_slot_list &＃61; []

　　　　for row in old_disk_list:

　　　　　　old_slot_list.append(row.slot)

　　update_list 交集更新&＃xff1b;create_list 差集创建&＃xff1b;del_list 差集删除

　　. . . . . .

六、API验证

1.自定义API验证

客户端代码

import time,requests,hashlibctime &＃61; time.time() # 获取当前时间戳
key &＃61; &＃39;asdfasdfasdfasdf098712sdfs&＃39;
new_key &＃61; &＃39;%s|%s&＃39;%(key,ctime)m &＃61; hashlib.md5()
m.update(bytes(new_key,encoding&＃61;&＃39;utf-8&＃39;))
md5_key &＃61; m.hexdigest()md5_time_key &＃61; &＃39;%s|%s&＃39;%(md5_key,ctime)
print(md5_time_key)
response &＃61; requests.get("http://127.0.0.1:8000/api/asset.html",headers&＃61;{&＃39;OpenKey&＃39;:md5_time_key})
print(response.text)

View Code

服务端

　　配置文件settings里加上 AUTH_KEY &＃61; "asdfasdfasdfasdf098712sdfs"

代码

api_key_record &＃61; {# "1b96b89695f52ec9de8292a5a7945e38|1501472467.4977243":1501472477.4977243
}
def asset(request):# 所有的信息放在 request.META 里client_md5_time_key &＃61; request.META.get(&＃39;HTTP_OPENKEY&＃39;)client_md5_key,client_ctime &＃61; client_md5_time_key.split(&＃39;|&＃39;)ctime &＃61; float(client_ctime)server_time &＃61; time.time()# 第一关if server_time - ctime > 10:return HttpResponse(&＃39;此随机字符串已失效&＃39;)# 第二关# settings.AUTH_KEYtmp &＃61; &＃39;%s|%s&＃39;%(settings.AUTH_KEY,client_ctime)m &＃61; hashlib.md5()m.update(bytes(tmp,encoding&＃61;&＃39;utf-8&＃39;))server_md5_key &＃61; m.hexdigest()if server_md5_key !&＃61; client_md5_key:return HttpResponse(&＃39;字符串中的时间不允许被修改&＃39;)# 维护 api_key_record 字典等容量for k in list(api_key_record.keys()):v &＃61; float(api_key_record[k])if server_time > v &＃43; 10 :del api_key_record[k]# 第三关if client_md5_time_key in api_key_record:return HttpResponse(&＃39;此随机字符串已访问过服务器端&＃39;)else:# 把随机字符串当做key&＃xff0c;把发过来时的时间当做value 记录进维护的api_key_record字典api_key_record[client_md5_time_key] &＃61; client_ctimeif request.method &＃61;&＃61; &＃39;GET&＃39;:import_info &＃61; &＃39;我是重要的数据&＃39;return HttpResponse(import_info)

View Code

2.AES数据加密

#3.6安装 pip3 install pycryptodome
#mac pip3 install pycrypto

安装

客户端

utils.py是封装数据加密、数据解密、API验证函数的组件

from Crypto.Cipher import AES
from lib.conf.config import settings# 数据加密
def encrypt(message):"""数据加密:param message::return:"""key &＃61; settings.DATA_KEYcipher &＃61; AES.new(key, AES.MODE_CBC, key)ba_data &＃61; bytearray(message,encoding&＃61;&＃39;utf-8&＃39;)v1 &＃61; len(ba_data)v2 &＃61; v1 % 16if v2 &＃61;&＃61; 0:v3 &＃61; 16else:v3 &＃61; 16 - v2for i in range(v3):ba_data.append(v3)final_data &＃61; ba_data.decode(&＃39;utf-8&＃39;)msg &＃61; cipher.encrypt(final_data) # 要加密的字符串&＃xff0c;必须是16个字节或16个字节的倍数return msg# 数据解密
def decrypt(msg):"""数据解密:param message::return:"""from Crypto.Cipher import AESkey &＃61; settings.DATA_KEYcipher &＃61; AES.new(key, AES.MODE_CBC, key)result &＃61; cipher.decrypt(msg) # result &＃61; b&＃39;\xe8\xa6\x81\xe5\x8a\xa0\xe5\xaf\x86\xe5\x8a\xa0\xe5\xaf\x86\xe5\x8a\xa0sdfsd\t\t\t\t\t\t\t\t\t&＃39;data &＃61; result[0:-result[-1]]return str(data,encoding&＃61;&＃39;utf-8&＃39;)# API验证
def auth():"""API验证:return:"""import hashlib,timectime &＃61; time.time()key &＃61; "asdfasdfasdfasdf098712sdfs"new_key &＃61; "%s|%s" %(key,ctime,)m &＃61; hashlib.md5()m.update(bytes(new_key,encoding&＃61;&＃39;utf-8&＃39;)) #里面是字节数据md5_key &＃61; m.hexdigest() #返回值是字符串类型
md5_time_key &＃61; "%s|%s" %(md5_key,ctime)return md5_time_key

utils.py

在client.py中的基类里&＃xff0c;把要发给服务端的API信息进行加密

import requests
from lib.conf.config import settings
from lib.utils import encrypt,auth
class Base(object):def post_asset(self,server_info):data &＃61; encrypt(json.dumps(server_info))requests.post(url&＃61;settings.API,data&＃61;data,headers&＃61;{&＃39;OpenKey&＃39;:auth(),&＃39;Content-Type&＃39;:&＃39;application/json&＃39;})

client.py

服务端

先弄出解密函数

def decrypt(msg):from Crypto.Cipher import AESkey &＃61; b&＃39;dfdsdfsasdfdsdfs&＃39;cipher &＃61; AES.new(key, AES.MODE_CBC, key)result &＃61; cipher.decrypt(msg) # result &＃61; b&＃39;\xe8\xa6\x81\xe5\x8a\xa0\xe5\xaf\x86\xe5\x8a\xa0\xe5\xaf\x86\xe5\x8a\xa0sdfsd\t\t\t\t\t\t\t\t\t&＃39;data &＃61; result[0:-result[-1]]return str(data,encoding&＃61;&＃39;utf-8&＃39;)