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