优化转存结果的查询逻辑和回显

This commit is contained in:
Cp0204 2024-04-14 01:08:07 +08:00
parent da35ebfa33
commit d49b7a6f18

View File

@ -11,6 +11,7 @@ import os
import re import re
import sys import sys
import json import json
import time
import random import random
import requests import requests
from datetime import datetime from datetime import datetime
@ -533,22 +534,37 @@ class Quark:
return False return False
def query_task(self, task_id): def query_task(self, task_id):
url = "https://drive-m.quark.cn/1/clouddrive/task" retry_index = 0
querystring = { while True:
"pr": "ucpro", url = "https://drive-m.quark.cn/1/clouddrive/task"
"fr": "pc", querystring = {
"uc_param_str": "", "pr": "ucpro",
"task_id": task_id, "fr": "pc",
"retry_index": "1", "uc_param_str": "",
"__dt": int(random.uniform(1, 5) * 60 * 1000), "task_id": task_id,
"__t": datetime.now().timestamp(), "retry_index": retry_index,
} "__dt": int(random.uniform(1, 5) * 60 * 1000),
headers = self.common_headers() "__t": datetime.now().timestamp(),
response = requests.request( }
"GET", url, headers=headers, params=querystring headers = self.common_headers()
).json() response = requests.request(
if response["code"] == 32003: "GET", url, headers=headers, params=querystring
response["message"] = "容量超限" ).json()
if response["data"]["status"] != 0:
if retry_index > 0:
print()
break
else:
if retry_index == 0:
print(
f"正在等待[{response['data']['task_title']}]执行结果",
end="",
flush=True,
)
else:
print(".", end="", flush=True)
retry_index += 1
time.sleep(0.500)
return response return response
def do_rename_task(self, task): def do_rename_task(self, task):