67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
|
import requests
|
|||
|
|
import json
|
|||
|
|
import hmac
|
|||
|
|
from hashlib import sha1
|
|||
|
|
import base64
|
|||
|
|
import time
|
|||
|
|
import uuid
|
|||
|
|
|
|||
|
|
from Config.Config import LIBLIB_SECRETKEY, LIBLIB_URL, LIBLIB_ACCESSKEY
|
|||
|
|
|
|||
|
|
|
|||
|
|
def make_sign(uri):
|
|||
|
|
"""
|
|||
|
|
生成签名
|
|||
|
|
"""
|
|||
|
|
# 当前毫秒时间戳
|
|||
|
|
timestamp = str(int(time.time() * 1000))
|
|||
|
|
# 随机字符串
|
|||
|
|
signature_nonce = str(uuid.uuid4())
|
|||
|
|
# 拼接请求数据
|
|||
|
|
content = '&'.join((uri, timestamp, signature_nonce))
|
|||
|
|
# 生成签名
|
|||
|
|
digest = hmac.new(LIBLIB_SECRETKEY.encode(), content.encode(), sha1).digest()
|
|||
|
|
# 移除为了补全base64位数而填充的尾部等号
|
|||
|
|
sign = base64.urlsafe_b64encode(digest).rstrip(b'=').decode()
|
|||
|
|
return sign, timestamp, signature_nonce
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 请求API接口的uri地址
|
|||
|
|
uri = "/api/model/version/get"
|
|||
|
|
Signature, timestamp, signature_nonce = make_sign(uri)
|
|||
|
|
url = f"{LIBLIB_URL}{uri}?AccessKey={LIBLIB_ACCESSKEY}&Signature={Signature}&Timestamp={timestamp}&SignatureNonce={signature_nonce}"
|
|||
|
|
|
|||
|
|
# 设置请求头
|
|||
|
|
headers = {"Content-Type": "application/json"}
|
|||
|
|
version_uuid = "4bb1335feb1e4d2eafe5a77bb93e861f"
|
|||
|
|
# 准备请求体数据
|
|||
|
|
request_body = {"version_uuid": version_uuid}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 发送POST请求
|
|||
|
|
response = requests.post(url, headers=headers, data=json.dumps(request_body))
|
|||
|
|
response.raise_for_status() # 检查请求是否成功
|
|||
|
|
print(response)
|
|||
|
|
# 解析响应数据
|
|||
|
|
result = response.json()
|
|||
|
|
print("API响应结果:")
|
|||
|
|
print(f"模型名称: {result.get('model_name')}")
|
|||
|
|
print(f"版本号: {result.get('version_name')}")
|
|||
|
|
print(f"是否可商用: {'是' if result.get('commercial_use') == '1' else '否'}")
|
|||
|
|
print(f"模型链接: {result.get('model_url')}")
|
|||
|
|
|
|||
|
|
except requests.exceptions.RequestException as e:
|
|||
|
|
print(f"请求发生错误: {str(e)}")
|
|||
|
|
"""
|
|||
|
|
{
|
|||
|
|
"version_uuid": "21df5d84cca74f7a885ba672b5a80d19",//LiblibAI官网模型链接后缀
|
|||
|
|
"model_name": "AWPortrait XL"
|
|||
|
|
"version_name": "1.1"
|
|||
|
|
"baseAlgo": "基础算法 XL"，
|
|||
|
|
"show_type": "1"，//公开可用的模型
|
|||
|
|
"commercial_use": "1"，//可商用为1，不可商用为0
|
|||
|
|
"model_url": "https://www.liblib.art/modelinfo/f8b990b20cb943e3aa0e96f34099d794?versionUuid=21df5d84cca74f7a885ba672b5a80d19"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
"""
|