gofile

commit ebc544e9729213f81655291b1dfd344047de871f
parent ec683b9a48739c062777df51badb61361bb12ad3
Author: archiveanon <>
Date:   Sun,  7 Apr 2024 09:00:42 +0000

Retry upload on connection error

Diffstat:
M
src/gofile/api.py
 | 
18
++++++++++++------

1 file changed, 12 insertions(+), 6 deletions(-)
diff --git a/src/gofile/api.py b/src/gofile/api.py
@@ -135,12 +135,18 @@ def upload_single(
             lambda monitor: progress.update(monitor.bytes_read - progress.n),
         )
 
-        upload_result = _gofile_api_post(
-            f"https://{server}.gofile.io/uploadFile",
-            data=monitor,
-            headers={"Content-Type": multipart_data.content_type},
-            type=GofileUploadResult,
-        )
+        while True:
+            try:
+                upload_result = _gofile_api_post(
+                    f"https://{server}.gofile.io/uploadFile",
+                    data=monitor,
+                    headers={"Content-Type": multipart_data.content_type},
+                    type=GofileUploadResult,
+                )
+                break
+            except (requests.exceptions.ConnectionError, ConnectionResetError) as e:
+                print(e)
+                pass
 
     return GofileUpload(file, upload_result)