From d08324e1590df4a7da19b70430e4abe364850d9b Mon Sep 17 00:00:00 2001
From: Andrew Godwin
Date: Sat, 17 Dec 2022 15:00:50 -0700
Subject: Handle timeouts on image fetch

---
 users/models/identity.py | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

(limited to 'users/models')

diff --git a/users/models/identity.py b/users/models/identity.py
index c8a154c..afec0e7 100644
--- a/users/models/identity.py
+++ b/users/models/identity.py
@@ -46,11 +46,15 @@ class IdentityStates(StateGraph):
         if await identity.fetch_actor():
             # Also stash their icon if we can
             if identity.icon_uri:
-                file, mimetype = await get_remote_file(
-                    identity.icon_uri,
-                    timeout=settings.SETUP.REMOTE_TIMEOUT,
-                    max_size=settings.SETUP.AVATAR_MAX_IMAGE_FILESIZE_KB * 1024,
-                )
+                try:
+                    file, mimetype = await get_remote_file(
+                        identity.icon_uri,
+                        timeout=settings.SETUP.REMOTE_TIMEOUT,
+                        max_size=settings.SETUP.AVATAR_MAX_IMAGE_FILESIZE_KB * 1024,
+                    )
+                except httpx.RequestError:
+                    # We've still got enough info to consider ourselves updated
+                    return cls.updated
                 if file:
                     identity.icon = file
                     await sync_to_async(identity.save)()
-- 
cgit v1.2.3