Browse Source

fix: site icon not showing (#11094)

Hash Brown 5 months ago
parent
commit
0f85e3557b
1 changed files with 2 additions and 2 deletions
  1. 2 2
      api/libs/helper.py

+ 2 - 2
api/libs/helper.py

@@ -31,12 +31,12 @@ class AppIconUrlField(fields.Raw):
         if obj is None:
         if obj is None:
             return None
             return None
 
 
-        from models.model import App, IconType
+        from models.model import App, IconType, Site
 
 
         if isinstance(obj, dict) and "app" in obj:
         if isinstance(obj, dict) and "app" in obj:
             obj = obj["app"]
             obj = obj["app"]
 
 
-        if isinstance(obj, App) and obj.icon_type == IconType.IMAGE.value:
+        if isinstance(obj, App | Site) and obj.icon_type == IconType.IMAGE.value:
             return file_helpers.get_signed_file_url(obj.icon)
             return file_helpers.get_signed_file_url(obj.icon)
         return None
         return None