open_pub/Uni-Lab-OS
1
0
Fork 0
mirror of https://github.com/dptech-corp/Uni-Lab-OS.git synced 2025-12-17 13:01:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

create/update resources with POST/PUT for big amount/ small amount data

Browse Source
This commit is contained in:
Junhan Chang
2025-09-26 23:25:34 +08:00
parent 622e579063
commit 14b7d52825
1 changed files with 33 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
unilabos/app/web/client.py
View File

@@ -25,6 +25,7 @@ class HTTPClient:
remote_addr: 远程服务器地址，如果不提供则从配置中获取 remote_addr: 远程服务器地址，如果不提供则从配置中获取
auth: 授权信息 auth: 授权信息
""" """
self.initialized = False
self.remote_addr = remote_addr or HTTPConfig.remote_addr self.remote_addr = remote_addr or HTTPConfig.remote_addr
if auth is not None: if auth is not None:
self.auth = auth self.auth = auth
@@ -69,12 +70,22 @@ class HTTPClient:
Returns: Returns:
Response: API响应对象 Response: API响应对象
""" """
if not self.initialized:
self.initialized = True
info(f"首次添加资源，当前远程地址: {self.remote_addr}")
response = requests.post( response = requests.post(
f"{self.remote_addr}/lab/material", f"{self.remote_addr}/lab/material",
json={"nodes": resources}, json={"nodes": resources},
headers={"Authorization": f"Lab {self.auth}"}, headers={"Authorization": f"Lab {self.auth}"},
timeout=100, timeout=100,
) )
else:
response = requests.put(
f"{self.remote_addr}/lab/material",
json={"nodes": resources},
headers={"Authorization": f"Lab {self.auth}"},
timeout=100,
)
if response.status_code == 200: if response.status_code == 200:
res = response.json() res = response.json()
if "code" in res and res["code"] != 0: if "code" in res and res["code"] != 0:
@@ -130,6 +141,16 @@ class HTTPClient:
Returns: Returns:
Response: API响应对象 Response: API响应对象
""" """
if not self.initialized:
self.initialized = True
info(f"首次添加资源，当前远程地址: {self.remote_addr}")
response = requests.post(
f"{self.remote_addr}/lab/material",
json={"nodes": resources},
headers={"Authorization": f"Lab {self.auth}"},
timeout=100,
)
else:
response = requests.put( response = requests.put(
f"{self.remote_addr}/lab/material", f"{self.remote_addr}/lab/material",
json={"nodes": resources}, json={"nodes": resources},