Skip to content

Commit b642fc3

Browse files
committed
implement upload retries
1 parent 57db0a1 commit b642fc3

File tree

1 file changed

+20
-5
lines changed

1 file changed

+20
-5
lines changed

ayon_api/server_api.py

Lines changed: 20 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1569,11 +1569,26 @@ def _upload_file(
15691569
if not chunk_size:
15701570
chunk_size = self.default_upload_chunk_size
15711571

1572-
response = post_func(
1573-
url,
1574-
data=self._upload_chunks_iter(stream, progress, chunk_size),
1575-
**kwargs
1576-
)
1572+
retries = self.get_default_max_retries()
1573+
response = None
1574+
for attempt in range(retries):
1575+
try:
1576+
response = post_func(
1577+
url,
1578+
data=self._upload_chunks_iter(
1579+
stream, progress, chunk_size
1580+
),
1581+
**kwargs
1582+
)
1583+
break
1584+
1585+
except (
1586+
requests.exceptions.Timeout,
1587+
requests.exceptions.ConnectionError,
1588+
):
1589+
if attempt == retries:
1590+
raise
1591+
progress.next_attempt()
15771592

15781593
response.raise_for_status()
15791594
return response

0 commit comments

Comments
 (0)