增加重试功能

This commit is contained in:
taskmgr818
2024-11-03 18:47:17 +08:00
parent c8c78a2c30
commit 97f9fcbbcb
4 changed files with 68 additions and 51 deletions

52
main.py
View File

@@ -8,7 +8,9 @@ import httpx
t = time.time()
tt = time.time()
reg = httpx.get(f"https://www.geetest.com/demo/gt/register-click-official?t={str(round(time.time()))}").json()
reg = httpx.get(
f"https://www.geetest.com/demo/gt/register-click-official?t={str(round(time.time()))}"
).json()
print(time.time() - tt)
crack = Crack(reg["gt"], reg["challenge"])
@@ -29,28 +31,32 @@ print(time.time() - tt)
model = Model()
tt = time.time()
pic_content = crack.get_pic()
print(time.time() - tt)
for retry in range(6):
tt = time.time()
pic_content = crack.get_pic(retry)
print(time.time() - tt)
tt = time.time()
small_img, big_img = model.detect(pic_content)
print(f"检测到小图: {len(small_img.keys())}个,大图: {len(big_img)} 个,用时: {time.time() - tt}s")
tt = time.time()
result_list = model.siamese(small_img, big_img)
print(f"文字配对完成,用时: {time.time() - tt}")
point_list = []
# print(result_list)
for i in result_list:
left = str(round((i[0] + 30) / 333 * 10000))
top = str(round((i[1] + 30) / 333 * 10000))
point_list.append(f"{left}_{top}")
wait_time = 4.0 - (time.time() - t)
time.sleep(wait_time)
tt = time.time()
result = json.loads(crack.verify(point_list))
print(result)
print(time.time() - tt)
ttt = tt = time.time()
small_img, big_img = model.detect(pic_content)
print(
f"检测到小图: {len(small_img.keys())}个,大图: {len(big_img)} 个,用时: {time.time() - tt}s"
)
tt = time.time()
result_list = model.siamese(small_img, big_img)
print(f"文字配对完成,用时: {time.time() - tt}")
point_list = []
# print(result_list)
for i in result_list:
left = str(round((i[0] + 30) / 333 * 10000))
top = str(round((i[1] + 30) / 333 * 10000))
point_list.append(f"{left}_{top}")
wait_time = 2.0 - (time.time() - ttt)
time.sleep(wait_time)
tt = time.time()
result = json.loads(crack.verify(point_list))
print(result)
print(time.time() - tt)
if result["data"]["result"] == "success":
break
total_time = time.time() - t
print(f"总计耗时(含等待{wait_time}s): {total_time}")