From c4cb993799454111979746387cc975470b3f6461 Mon Sep 17 00:00:00 2001
From: wizardforcel <562826179@qq.com>
Date: Sat, 7 Mar 2020 14:19:14 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E8=B6=85=E6=97=B6=E6=9C=BA?=
 =?UTF-8?q?=E5=88=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 BiliDriveEx/util.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/BiliDriveEx/util.py b/BiliDriveEx/util.py
index 3ae1e55..7e179a3 100644
--- a/BiliDriveEx/util.py
+++ b/BiliDriveEx/util.py
@@ -32,8 +32,11 @@ def image_download(url):
     content = []
     last_chunk_time = None
     try:
-        for chunk in requests.get(url, headers=headers, timeout=10, stream=True).iter_content(128 * 1024):
-            if last_chunk_time is not None and time.time() - last_chunk_time > 5:
+        for chunk in request_retry('GET', url, 
+            headers=headers, stream=True
+        ).iter_content(128 * 1024):
+            if last_chunk_time is not None and \
+               time.time() - last_chunk_time > 5:
                 return
             content.append(chunk)
             last_chunk_time = time.time()
@@ -71,6 +74,7 @@ def log(message):
     print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] {message}")
     
 def request_retry(method, url, retry=5, **kwargs):
+    kwargs.setdefault('timeout', 10)
     for i in range(retry):
         try:
             return requests.request(method, url, **kwargs)
-- 
GitLab