gofile

commit 56a10d0ee18455775a809e9de29e92411236054b
parent 7bdc3c0cc10624f94e81b8ecca8b2f6af023c131
Author: archiveanon <>
Date:   Mon,  8 Apr 2024 04:36:04 +0000

Fix retry handling

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

1 file changed, 19 insertions(+), 19 deletions(-)
diff --git a/src/gofile/api.py b/src/gofile/api.py
@@ -145,22 +145,22 @@ def upload_single(
     )
 
     file_size = file.stat().st_size
-    with tqdm.tqdm(
-        desc=short_file_name,
-        total=file_size,
-        unit="B",
-        unit_scale=True,
-        unit_divisor=1024,
-        dynamic_ncols=True,
-    ) as progress:
-        multipart_data = MultipartEncoder(fields=post_data)
-        monitor = MultipartEncoderMonitor(
-            multipart_data,
-            lambda monitor: progress.update(monitor.bytes_read - progress.n),
-        )
-
-        while True:
-            try:
+    while True:
+        try:
+            with tqdm.tqdm(
+                desc=short_file_name,
+                total=file_size,
+                unit="B",
+                unit_scale=True,
+                unit_divisor=1024,
+                dynamic_ncols=True,
+            ) as progress:
+                multipart_data = MultipartEncoder(fields=post_data)
+                monitor = MultipartEncoderMonitor(
+                    multipart_data,
+                    lambda monitor: progress.update(monitor.bytes_read - progress.n),
+                )
+
                 upload_result = _gofile_api_post(
                     f"https://{server}.gofile.io/contents/uploadfile",
                     data=monitor,
@@ -168,9 +168,9 @@ def upload_single(
                     type=GofileUploadResult,
                 )
                 break
-            except (requests.exceptions.ConnectionError, ConnectionResetError) as e:
-                print(e)
-                pass
+        except (requests.exceptions.ConnectionError, ConnectionResetError) as e:
+            print(e)
+            pass
 
     return GofileUpload(file, upload_result)