­

CMDB学习之六 –客户端请求测试,服

  • 2020 年 2 月 10 日
  • 筆記

客户端使用agent 请求测试，agent使用的POST 请求，使用requests模块

本地采集，汇报服务端

#!/usr/bin/env python  # -*- coding:utf-8 -*-  from .base import BaseHandler  from ..plugins import get_server_info  import requests  import json    class AgnetHandler(BaseHandler):        def cmd(self,command,hostname=None):          import subprocess          return subprocess.getoutput(command)        def handler(self):          """          Agent模式下处理资产采集:硬盘、内存、网卡          :return:          """          print('agent模式')          #调用pulugins.disk /plugins.momory /plugins.nerwork          info = get_server_info(self)          # print(info)          r1 = requests.post(              url='http://127.0.0.1:8000/api/asset',              data=json.dumps(info)          )          print(r1)

salt 和ssh请求 首先要获取服务端 ，没有采集的主机列表，然后根据主机列表进行采集。

　　1. get请求获取主机列表

　　2、发起远程采集数据，汇报服务器

两方法都有 采集都发起远程和汇报服务器的功能，所有在基类实现方法，然后 方法继承基类   SSHhandSaltHandler

#!/usr/bin/env python  # -*- coding:utf-8 -*-  from ..plugins import get_server_info  import json  import requests    class BaseHandler(object):        def cmd(self,command,hostname=None):          '''           约束派生类          :return:          '''          raise NotImplementedError('cmd() must Implemented.')        def handler(self):          '''          约束派生类          :return:          '''          raise NotImplementedError('handler() must Implemented.')    class SSHhandSaltHandler(BaseHandler):      def handler(self):          """          Salt模式下处理资产采集          :return:          """          print('salt模式或者ssh模式')          r1 = requests.get(url='http://127.0.0.1:8000/api/asset')          host_list = r1.json()          print(host_list)          from concurrent.futures import ThreadPoolExecutor          pool = ThreadPoolExecutor(10)          for host in host_list:              pool.submit(self.task, host)        def task(self, hostname):          # 资产采集并且汇报          info = get_server_info(self)          # print(info)          r1 = requests.post(              url='http://127.0.0.1:8000/api/asset',              data=json.dumps(info)          )          print(r1)          print(r1.text)
#!/usr/bin/env python  # -*- coding:utf-8 -*-  from .base import SSHhandSaltHandler    class SaltHandler(SSHhandSaltHandler):        def cmd(self, command, hostname=None):          """          调用saltstack远程连接主机并执行命令（saltstack的master）          :param hostname:主机名          :param command: 要执行的命令          :return:          """          import salt.client #安装salt          local = salt.client.LocalClient()          result = local.cmd(hostname, 'cmd.run', [command])          return result[hostname]
#!/usr/bin/env python  # -*- coding:utf-8 -*-  from .base import SSHhandSaltHandler  from config  import settings    class SSHHandler(SSHhandSaltHandler):        def cmd(self, command, hostname=None):          """          调用paramiko远程连接主机并执行命令，依赖rsa          :param hostname:主机名          :param command: 要执行的命令          :return:          """          import paramiko            private_key = paramiko.RSAKey.from_private_key_file(settings.SSH_PRIVATE_KEY)          ssh = paramiko.SSHClient()          ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())          ssh.connect(hostname=hostname, port=settings.SSH_PORT, username=settings.SSH_USER, pkey=private_key)          stdin, stdout, stderr = ssh.exec_command(command)          result = stdout.read()          ssh.close()          return result

##################下面是django服务端#######################

下面使用多种方式的api 

from django.shortcuts import render,HttpResponse,redirect,reverse  from django.views.decorators.csrf import csrf_exempt  import json  # Create your views here.  @csrf_exempt  def asset(request):      if request.method == 'GET':          host_list = ['c1.com','c2.com']          return HttpResponse(json.dumps(host_list)) #序列化        # print(request.body) #原始数据      # print(request.POST) #url编码的格式  "hostname=c1.com&memory=...."      ret =json.loads(request.body.decode('utf-8'))      print(ret)      return HttpResponse("接受成功")      # CBV加装饰器，忽略csrf验证  # from django.utils.decorators import method_decorator  # from django.views import View  # # @method_decorator(csrf_exempt,name='dispatch')  #这个装饰器比较特殊不能直接加到类中的方法，方法一  # class Asset(View):  #  #     # 这个装饰器比较特殊不能直接加到类中的方法，方法二  #     @method_decorator(csrf_exempt)  #     def dispatch(self,request,*args,**kwargs):  #         return super().dispatch(request,*args,**kwargs)  #  #     def get(self,request):  #         host_list = ['c1.com', 'c2.com']  #         return HttpResponse(json.dumps(host_list))  # 序列化  #  #     def post(self,request):  #         ret = json.loads(request.body.decode('utf-8'))  #         print(ret)  #         return HttpResponse("接受成功")    #使用rest_framework ，首先要安装pip去安装Djangorestframework ，这个模块  # 在Django的settings中注册app  from rest_framework.views import APIView  from rest_framework.response import Response    class Asset(APIView):        def get(self,request):          host_list = ['c1.com', 'c2.com']          return Response(host_list)        def post(self,request):          ret = json.loads(request.body.decode('utf-8'))          print(ret)          return HttpResponse("接受成功")

Django提供的 rest_framework 需要安装模块导入使用，方便简洁，rest_framework是一个独立的app，所以在Django配置文件中注册

页面结果;

VirMach 便宜 VPS

QNews

QNews