fix: site icon not showing (#11094)

1 year ago · 0f85e3557b
parent 17ee731546
commit 0f85e3557b
1 changed files with 2 additions and 2 deletions
--- a/api/libs/helper.py
+++ b/api/libs/helper.py
@ -31,12 +31,12 @@ class AppIconUrlField(fields.Raw):
        if obj is None:
            return None

-        from models.model import App, IconType
+        from models.model import App, IconType, Site

        if isinstance(obj, dict) and "app" in obj:
            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 None