From 61e54c3b5fb0f92fd5a1cdd42be0f7da727682b5 Mon Sep 17 00:00:00 2001
From: Sebastian Goscik <sebastian.goscik@live.co.uk>
Date: Mon, 21 Feb 2022 10:41:06 +0000
Subject: [PATCH] Add known download exception

---
 unifi_protect_backup/unifi_protect_backup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/unifi_protect_backup/unifi_protect_backup.py b/unifi_protect_backup/unifi_protect_backup.py
index 70e1779..d2b68ac 100644
--- a/unifi_protect_backup/unifi_protect_backup.py
+++ b/unifi_protect_backup/unifi_protect_backup.py
@@ -6,6 +6,7 @@ import shutil
 from typing import Callable, List, Optional
 
 import aiocron
+import aiohttp
 from pyunifiprotect import ProtectApiClient
 from pyunifiprotect.data.nvr import Event
 from pyunifiprotect.data.types import EventType, ModelType
@@ -372,7 +373,7 @@ class UnifiProtectBackup:
                         video = await self._protect.get_camera_video(event.camera_id, event.start, event.end)
                         assert isinstance(video, bytes)
                         break
-                    except AssertionError as e:
+                    except (AssertionError, aiohttp.client_exceptions.ClientPayloadError) as e:
                         logger.warn("    Failed download attempt {x+1}, retying in 1s")
                         logger.exception(e)
                         await asyncio.sleep(1)