CMDB学习之六 --客户端请求测试,服务端api优化

客户端使用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配置文件中注册

页面结果;

CMDB学习之六 --客户端请求测试,服务端api优化的相关教程结束。