当前位置:AIGC资讯 > AIGC > 正文

基于文心一言AI大模型,编写一段python3程序以获取华为分布式块存储REST接口的实时数据

本文尝试基于文心一言AI大模型,编写一段python3程序以获取华为分布式块存储REST接口的实时数据。

一、用文心一言AI大模型将需求转化为样例代码

1、第一次对话:“python3写一段从rest服务器获取数据的样例代码”

同时生成了以下注解 

这段代码首先定义了一个函数 get_data_from_rest_server,该函数使用 requests.get 方法向指定的 URL 发送 GET 请求。如果请求成功(即响应的状态码为 200),则将响应的 JSON 数据返回。否则,它将打印出错误消息并返回 None。然后,我们调用这个函数来获取数据,并打印出获取到的数据。注意你需要将 url 替换为你的 REST 服务器的实际 URL。 

  2、第二次对话:“以上请求需带入服务器的认证参数”

同时生成了以下注解 

 在上述代码中,我们通过在 requests.get 方法中传递 auth=(username, password) 来提供服务器的认证参数。这样,当你发送请求时,requests 会自动将认证参数添加到请求中,从而完成身份验证过程。

3、第三次对话:“以上代码需代入请求消息头”

import requests  
  
# 定义服务器认证参数的用户名和密码  
username = 'your_username'  
password = 'your_password'  
  
def get_data_from_rest_server():  
    # 将此 URL 替换为你的 REST 服务器 URL  
    url = 'http://example.com/api/data'  
  
    # 定义请求头,可以包含自定义的头部字段和值  
    headers = {  
        'Content-Type': 'application/json',  # 声明请求的内容类型为 JSON  
        'Authorization': f'Basic {username}:{password}',  # 添加基本的身份验证头部字段  
    }  
  
    # 使用 GET 方法请求数据,并携带服务器认证参数和请求头  
    response = requests.get(url, headers=headers)  
  
    # 检查响应的状态码,如果请求成功(状态码为 200),则处理响应的数据  
    if response.status_code == 200:  
        data = response.json()  
        return data  
    else:  
        print(f"请求失败,状态码: {response.status_code}")  
        return None  
  
# 调用函数获取数据  
data = get_data_from_rest_server()  
if data is not None:  
    print(data)

注解如下 :

在上述代码中,我们通过创建一个字典对象来定义请求头,其中包含了一个 Content-Type 字段和一个 Authorization 字段。Content-Type 字段声明了请求的内容类型为 JSON,而 Authorization 字段包含了基本的身份验证值。然后,我们将这个字典对象传递给 requests.get 方法的 headers 参数,从而将请求头添加到请求中。 

 通过和文心一言AI大模型的三次对话,我们大致了解了用python 的 requests 库从 REST 服务器获取数据的基本方法。

二、核查华为分布式块存储REST接口规范

1、RESTful API的请求必须包括如下消息头

2、拟测试请求接口官方样例

查询存储池 接口描述

查询存储池。

URI 路径

https://${ip}:${port}/dsware/service/cluster/storagepool/queryStoragePool

访问方法

GET

执行实例

Request: /dsware/service/cluster/storagepool/queryStorageNodeInfo?poolId=0

Response: {

"result": 0,

"storagePools": [

{

"poolId": 0,

"totalCapacity": 675305,

"usedCapacity": 0,

"reductionInvolvedCapacity": 0,

"allocatedCapacity": 0,

"usedCapacityRate": 0.0,

"deduplicationSaved": 0,

"compressionSaved": 0,

"deduplicationRatio": 1.0,

"compressionRatio": 1.0,

"dataReductionRatio": 1.0,

"thinRate": 0,

"replicationFactor": 3,

"poolName": "rep3",

"poolServerType": 1,

"thinThreshold": 70,

"poolSpec": "normal_pool",

"redundancyPolicy": "replication",

"numDataUnits": 0,

"numParityUnits": 0,

"numFaultTolerance": 0,

"cellSize": 0,

"ecCacheMediaType": "false",

"ecCacheRate": "0",

"ecCacheWbCacheRate": "0",

"compressionAlgorithm": "performance",

"enableAdvanceVolume": "true",

"encryptType": 0,

"supportEncryptForMainStorageMedia": 0,

"serviceType": 1,

"storageMediaType": "ssd_card",

"physicalTotalCapacity": 482361,

"usedCapacityAfterDedup": 0,

"writableCapacity": 482361,

"markDelCapacity": 0,

"bbuInfo": "close",

"fastGcTime": 0

}

]

}

三、代码实写

 根据需求进行代码编写,并多次测试,最终通过的代码如下(以下代码存为名为“healthck.py”的文件):

#!/usr/bin/python3
#coding=utf-8

##__author__='daijianbing'

import requests
import json
  
def authenticate(username, password,url_auth):
    # 构建认证数据
    auth_data = {
        'user_name': username,
        'password': password
    }
  
    # 将认证数据转换为JSON格式
    auth_json = json.dumps(auth_data)
  
    #禁止未经验证的HTTPS
    requests.packages.urllib3.disable_warnings()
    # 发送POST请求到REST API进行身份验证
    response = requests.post(url_auth,verify = False, data=auth_json, headers={'Content-Type': 'application/json'})
  
    # 检查响应状态码,如果成功则返回认证令牌,否则抛出异常
    if response.status_code == 200:
        #print(response.json())
        return response.json()['data']['x_auth_token']
    else:
        connect = 0
        raise Exception('Authentication failed')
  
def queryStoragePool(token,url):
    # 构建认证数据
    headers = {
        'Connection': 'keep-alive',
        'X-Auth-Token': token
    }

    #print(headers)

    # 发送请求到REST API
    response = requests.get(url, verify = False, headers=headers)

    # 检查响应状态码,如果成功则返回认证令牌,否则抛出异常
    if response.status_code == 200:
        #print(response.json())
        return response.json()['storagePools']
    else:
        raise Exception('Authentication failed')

def queryPoolstatus(token,url,poolId):
    # 构建认证数据
    headers = {
        'Connection': 'keep-alive',
        'X-Auth-Token': token
    }
    params = {
        'poolId':str(poolId),
        'vbs': 'true'
    }

    params_json = json.dumps(params)
    #print("url:",url,"headers:",headers,"params_json:",params_json)

    # 发送T请求到REST API
    response = requests.get(url, verify = False, params=params, headers=headers)

    # 检查响应状态码,如果成功则返回认证令牌,否则抛出异常
    if response.status_code == 200:
        #print(response.json())
        return response.json()['status']
    else:
        raise Exception('Authentication failed')

def queryAlarms(token,url):
    # 构建认证数据
    headers = {
        'Connection': 'keep-alive',
        'X-Auth-Token': token
    }
    params = {
        'filter':"alarmStatus::1",
    }
    params_json = json.dumps(params)
    # 发送请求到REST API
    response = requests.get(url, verify = False, params=params, headers=headers)

    # 检查响应状态码,如果成功则返回认证令牌,否则抛出异常
    if response.status_code == 200:
        #print(response.json())
        return response.json()['data']
    else:
        raise Exception('Authentication failed')

# 使用示例  
username = 'zabbix'  
password = 'passwd'
hostip = '192.168.0.1'
port = '8088'
hostport = "https://"+hostip+":"+port
url_auth = hostport+'/api/v2/aa/sessions'
checkstatus = 1
requests.packages.urllib3.disable_warnings()
try:
    token = authenticate(username, password, url_auth)
    print('x_auth_token:', token)

    url_pool = hostport+'/dsware/service/resource/queryStoragePool'
    pools = queryStoragePool(token,url_pool)

    url_poolstatus=hostport+'/dsware/service/cluster/storagepool/queryStatus'
    for pool in pools:
       poolstatus=queryPoolstatus(token,url_poolstatus,pool['poolId'])

    url_alarms = hostport+'/api/v2/common/alarm_count'
    Alarms=queryAlarms(token,url_alarms)
    print('poolid:',pool['poolId'],'poolname:',pool['poolName'],'poolstatus:',poolstatus,'alarmcount:',Alarms['count'])
except:
    checkstatus = 0
finally:
    print("checkstatus:",checkstatus)
    print("chech over")

四、验证运行

实际运行:

# python3 --version
Python 3.9.9
# python3 healthck.py   
x_auth_token: MzMwMDE0OTkyNkEzMDAxNGUwOTcwMzE2
poolid: 0 poolname: Pool01 poolstatus: 0 alarmcount: 0
checkstatus: 1
chech over

本段代码实现了存储REST接口认证登录,查询全部的pool并列出poolid,poolname,pool状态,输出当前未恢复告警数并返回本次的检查状态(1为检查成功)。

更新时间 2023-11-29