provide timeout on final download complete

gilesknap · Feb 24, 2019 · 1cbda53 · 1cbda53
1 parent d68c16d
commit 1cbda53
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gphotos/GooglePhotosDownload.py b/gphotos/GooglePhotosDownload.py
@@ -243,7 +243,9 @@ def do_download_complete(self,
         """
         for future in futures_list:
             media_item = self.pool_future_to_media.get(future)
-            e = future.exception()
+            timeout = self.video_timeout if media_item.is_video() else \
+                self.image_timeout
+            e = future.exception(timeout=timeout)
             if e:
                 self.files_download_failed += 1
                 log.error('FAILURE %d downloading %s',