Merge pull request #3 from caruhome/livioso-patch-retry-on-remote-disc

fix: retry on RemoteDisconnected
caruhome · Jan 13, 2020 · 3949ad2 · 3949ad2
2 parents 2c56e77 + efaa691
commit 3949ad2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/upparat/statemachine/download.py b/src/upparat/statemachine/download.py
@@ -5,6 +5,7 @@
 import urllib.request
 from urllib.error import HTTPError
 from urllib.error import URLError
+from http.client import RemoteDisconnected
 
 import backoff
 import pysm
@@ -25,7 +26,7 @@
 
 
 @backoff.on_exception(
-    backoff.expo, (URLError, HTTPError, socket.timeout), jitter=backoff.full_jitter
+    backoff.expo, (URLError, HTTPError, RemoteDisconnected, socket.timeout), jitter=backoff.full_jitter
 )
 def download(job, stop_download, publish, update_job_progress):
     start_position_bytes = 0