retry

get_version.py

@@ -25,12 +25,17 @@ GLOBAL_REQUEST_DELAY = 0.5
 
 def get_js_url() -> str:
     """请求百度网盘下载页,获取存在下载链接的js文件"""
+    max_attempts = 5
+    attempt = 0
+
+    url = ""
+    while attempt < max_attempts:
+        try:
             r = requests.get(
                 "https://pan.baidu.com/download",
                 headers=GLOBAL_HEADER,
             )
             e = etree.HTML(r.text, parser=None)
-    url = ""
 
             # 获取包含链接的js文件
             for i in e.xpath("//script/@src"):
@@ -38,6 +43,14 @@ def get_js_url() -> str:
                     url = i
             # print(url)
             return url
+        except Exception as e:
+            print(f"An error occurred: {e}")
+            attempt += 1
+            time.sleep(1)  # 等待1秒再重试
+
+    if attempt == max_attempts:
+        print("Failed after several attempts")
+    return url
 
 
 def append_json(json_file_name: str, add):
@@ -79,7 +92,13 @@ def parse_url_from_js():
     """从js文件中提取下载的链接并且去重,写入文件"""
     url = get_js_url()
     u_url = []
+
+    max_attempts = 5
+    attempt = 0
+
+    while attempt < max_attempts:
         try:
+
             d_url = ""
             r = requests.get(
                 url,
@@ -99,8 +118,15 @@ def parse_url_from_js():
             for i in d_url:
                 if i not in u_url:
                     u_url.append(i)
+
+            break
         except Exception as e:
-        print(e)
+            print(f"An error occurred: {e}")
+            attempt += 1
+            time.sleep(1)  # 等待1秒再重试
+
+    if attempt == max_attempts:
+        print("Failed after several attempts,set url to default")
         u_url = GLOBAL_VERSION
 
     with open("url.json", "w") as f:
@@ -176,6 +202,8 @@ def check_version(version):
     if attempt == max_attempts:
         print("Failed after several attempts")
 
+    return
+
 
 def find_latest_version(min_version, max_version):
     """寻找最新版本号"""