From 3e53d43f955aaad9153271208675727b50d89f85 Mon Sep 17 00:00:00 2001
From: Sebastian Goscik <sebastian.goscik@live.co.uk>
Date: Tue, 22 Feb 2022 22:57:50 +0000
Subject: [PATCH] Add timeout to known download exceptions

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

diff --git a/unifi_protect_backup/unifi_protect_backup.py b/unifi_protect_backup/unifi_protect_backup.py
index 171b278..a2201a9 100644
--- a/unifi_protect_backup/unifi_protect_backup.py
+++ b/unifi_protect_backup/unifi_protect_backup.py
@@ -3,10 +3,11 @@ import asyncio
 import logging
 import pathlib
 import shutil
+from asyncio.exceptions import TimeoutError
 from typing import Callable, List, Optional
 
 import aiocron
-import aiohttp
+from aiohttp.client_exceptions import ClientPayloadError
 from pyunifiprotect import NvrError, ProtectApiClient
 from pyunifiprotect.data.nvr import Event
 from pyunifiprotect.data.types import EventType, ModelType
@@ -380,7 +381,7 @@ class UnifiProtectBackup:
                         video = await self._protect.get_camera_video(event.camera_id, event.start, event.end)
                         assert isinstance(video, bytes)
                         break
-                    except (AssertionError, aiohttp.client_exceptions.ClientPayloadError) as e:
+                    except (AssertionError, ClientPayloadError, TimeoutError) as e:
                         logger.warn(f"    Failed download attempt {x+1}, retying in 1s")
                         logger.exception(e)
                         await asyncio.sleep(1)