From 4be2962faaba24ded023d820f54cafffa020b067 Mon Sep 17 00:00:00 2001
From: Jerry Yan <792602257@qq.com>
Date: Sat, 1 Oct 2022 14:53:11 +0800
Subject: [PATCH] =?UTF-8?q?=E9=81=BF=E5=85=8D=E6=BC=8F=E4=BC=A0chunk?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 workflow/bilibiliupload/core.py       | 4 ++--
 workflow/bilibiliupload/util/retry.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/workflow/bilibiliupload/core.py b/workflow/bilibiliupload/core.py
index d1d37b0..414bd58 100644
--- a/workflow/bilibiliupload/core.py
+++ b/workflow/bilibiliupload/core.py
@@ -241,8 +241,8 @@ def upload_chunk(upload_url, server_file_name, local_file_name, chunk_data, chun
     )
     print(r.status_code)
     print(r.content)
-
-    if r.status_code == 200 and r.json()['OK'] == 1:
+    r.raise_for_status()
+    if r.status_code == 200 and r.json().get("OK", 0) == 1:
         return True
     else:
         return False
diff --git a/workflow/bilibiliupload/util/retry.py b/workflow/bilibiliupload/util/retry.py
index d9f3151..c34d308 100644
--- a/workflow/bilibiliupload/util/retry.py
+++ b/workflow/bilibiliupload/util/retry.py
@@ -11,7 +11,7 @@ class Retry:
             try:
                 return_value = func(*args, **kwargs)
             except Exception:
-                return_value = not self.success_return_value
+                continue
             if return_value == self.success_return_value:
                 status = True
                 break