Skip to content

Commit c98d385

Browse files
committed
feat: retry on ConnectionResetError
1 parent 5fac423 commit c98d385

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/icloudpd/download.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def download_media(
125125
)
126126
break
127127

128-
except (ConnectionError, socket.timeout, PyiCloudAPIResponseException, Timeout) as ex:
128+
except (ConnectionError, ConnectionResetError, socket.timeout, PyiCloudAPIResponseException, Timeout) as ex:
129129
if "Invalid global session" in str(ex):
130130
logger.error("Session error, re-authenticating...")
131131
if retries > 0:

src/pyicloud_ipd/services/photos.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,7 @@ def photos(self) -> Generator["PhotoAsset", Any, None]:
410410
while(True):
411411
try:
412412
request = self.photos_request(offset)
413-
except PyiCloudAPIResponseException as ex:
413+
except (ConnectionError, ConnectionResetError, PyiCloudAPIResponseException) as ex:
414414
if self.exception_handler:
415415
exception_retries += 1
416416
self.exception_handler(ex, exception_retries)

0 commit comments

Comments
 (0)